Carries information about input or output of the algorithm. IAlgorithmIOInfo for all the input and output along with IAlgorithmVariant denotes the variation of algorithm and can be used to select or reproduce an algorithm using IAlgorithmSelector::selectAlgorithms().
More...
#include <NvInfer.h>
|
apiv::VAlgorithmIOInfo * | mImpl |
|
Carries information about input or output of the algorithm. IAlgorithmIOInfo for all the input and output along with IAlgorithmVariant denotes the variation of algorithm and can be used to select or reproduce an algorithm using IAlgorithmSelector::selectAlgorithms().
- See also
- IAlgorithmVariant, IAlgorithm, IAlgorithmSelector::selectAlgorithms()
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
◆ ~IAlgorithmIOInfo()
virtual nvinfer1::IAlgorithmIOInfo::~IAlgorithmIOInfo |
( |
| ) |
|
|
protectedvirtualdefaultnoexcept |
◆ getComponentsPerElement()
int64_t nvinfer1::IAlgorithmIOInfo::getComponentsPerElement |
( |
| ) |
const |
|
inlinenoexcept |
Return the number of components per element. This is always 1 for non-vectorized formats.
- Returns
- the number of components per element.
◆ getDataType()
DataType nvinfer1::IAlgorithmIOInfo::getDataType |
( |
| ) |
const |
|
inlinenoexcept |
Return DataType of the input/output of algorithm.
- Returns
- the data type.
◆ getStrides()
Dims nvinfer1::IAlgorithmIOInfo::getStrides |
( |
| ) |
const |
|
inlinenoexcept |
Return strides of the input/output tensor of algorithm. For vectorized formats, strides are given in units of vectors.
- Returns
- the strides of the tensor.
◆ getTensorFormat()
Return TensorFormat of the input/output of algorithm.
- Deprecated:
- Deprecated in TensorRT 8.6. The strides, data type, and vectorization information is sufficient to uniquely identify tensor formats.
- Returns
- the tensor format
◆ getVectorizedDim()
int64_t nvinfer1::IAlgorithmIOInfo::getVectorizedDim |
( |
| ) |
const |
|
inlinenoexcept |
Return the index of the vectorized dimension or -1 for non-vectorized formats.
- Returns
- the index of the vectorized dimension.
◆ mImpl
apiv::VAlgorithmIOInfo* nvinfer1::IAlgorithmIOInfo::mImpl |
|
protected |
The documentation for this class was generated from the following file: