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>
|
DataType | getDataType () const noexcept |
| Return DataType of the input/output of algorithm. More...
|
|
Dims | getStrides () const noexcept |
| Return strides of the input/output tensor of algorithm. For vectorized formats, strides are given in units of vectors. More...
|
|
int64_t | getVectorizedDim () const noexcept |
| Return the index of the vectorized dimension or -1 for non-vectorized formats. More...
|
|
int64_t | getComponentsPerElement () const noexcept |
| Return the number of components per element. This is always 1 for non-vectorized formats. More...
|
|
|
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.
◆ 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: