TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

How to implement a tensorrt plugin for "Unique" op?

Open BobLiu20 opened this issue 1 year ago • 1 comments

I want to implement a plug-in for the operator Unique. But I don’t know how to determine its output dimension, because it cannot be determined directly based on the input (must be determined after calculation? )

nvinfer1::DataType UniquePlugin::getOutputDataType(int index, const nvinfer1::DataType* inputTypes, int nbInputs) const
    noexcept
{
    ?????????????
}

Numpy "Unuqie": https://numpy.org/doc/stable/reference/generated/numpy.unique.html

Any idea to implement it?

BobLiu20 avatar Feb 07 '24 02:02 BobLiu20

Currently we don't support data dependent plugin. @samurdhikaru for viz.

zerollzeng avatar Feb 08 '24 06:02 zerollzeng

closing as duplicate with https://github.com/NVIDIA/TensorRT/issues/3521, we need DDS plugin support.

ttyio avatar Apr 16 '24 18:04 ttyio