TensorRT 8.6.1
|
Fields that a plugin might see for an input or output. More...
#include <NvInferRuntimePlugin.h>
Public Attributes | |
Dims | dims |
Dimensions. More... | |
DataType | type |
TensorFormat | format |
Tensor format. More... | |
float | scale |
Scale for INT8 data type. More... | |
Fields that a plugin might see for an input or output.
Scale is only valid when data type is DataType::kINT8. TensorRT will set the value to -1.0f if it is invalid.
Dims nvinfer1::PluginTensorDesc::dims |
Dimensions.
TensorFormat nvinfer1::PluginTensorDesc::format |
Tensor format.
float nvinfer1::PluginTensorDesc::scale |
Scale for INT8 data type.
DataType nvinfer1::PluginTensorDesc::type |