TensorRT  7.1.3.0
nvinfer1::PluginTensorDesc Struct Reference

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

#include <NvInferRuntimeCommon.h>

Public Attributes

Dims dims
 
DataType type
 
TensorFormat format
 
float scale
 

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::supportsFormat
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: