TensorRT 10.0.1
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

Member Data Documentation

◆ dims

Dims nvinfer1::PluginTensorDesc::dims


◆ format

TensorFormat nvinfer1::PluginTensorDesc::format

Tensor format.

◆ scale

float nvinfer1::PluginTensorDesc::scale

Scale for INT8 data type.

◆ type

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

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

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact