nvinfer1::PluginTensorDesc Struct Reference

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

Collaboration diagram for nvinfer1::PluginTensorDesc:

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

dims

Dims nvinfer1::PluginTensorDesc::dims

type

DataType nvinfer1::PluginTensorDesc::type

Warning

DataType:kBOOL not supported.

format

TensorFormat nvinfer1::PluginTensorDesc::format

scale

float nvinfer1::PluginTensorDesc::scale

---

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

• NvInferRuntimeCommon.h