TensorRT 8.5.3
nvinfer1::PluginTensorDesc Struct Reference

Fields that a plugin might see for an input or output. More...

#include <NvInferRuntimeCommon.h>

Public Attributes

Dims dims
 Dimensions. More...
 
DataType type
 
TensorFormat format
 Tensor format. More...
 
float scale
 Scale for INT8 data type. More...
 

Detailed Description

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.

See also
IPluginV2IOExt::supportsFormatCombination
IPluginV2IOExt::configurePlugin

Member Data Documentation

◆ dims

Dims nvinfer1::PluginTensorDesc::dims

Dimensions.

◆ format

TensorFormat nvinfer1::PluginTensorDesc::format

Tensor format.

◆ scale

float nvinfer1::PluginTensorDesc::scale

Scale for INT8 data type.

◆ type

DataType nvinfer1::PluginTensorDesc::type
Warning
DataType:kBOOL and DataType::kUINT8 are not supported.

The documentation for this struct was generated from the following file: