nvinfer1::PluginTensorDesc Struct Reference

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...

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:

• NvInferRuntimePlugin.h