TensorRT
TensorRT copied to clipboard
How to implement a tensorrt plugin for "Unique" op?
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?
Currently we don't support data dependent plugin. @samurdhikaru for viz.
closing as duplicate with https://github.com/NVIDIA/TensorRT/issues/3521, we need DDS plugin support.