TensorRT  8.0.2
nvinfer1::PluginTensorDesc Struct Reference

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

#include <NvInferRuntimeCommon.h>

Public Attributes

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

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

◆ type

DataType nvinfer1::PluginTensorDesc::type
Warning
DataType:kBOOL not supported.

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