TensorRT
7.1.3.0
|
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>
Public Member Functions | |
virtual TensorFormat | getTensorFormat () const =0 |
Return TensorFormat of the input/output of algorithm. | |
virtual DataType | getDataType () const =0 |
Return DataType of the input/output of algorithm. | |
virtual Dims | getStrides () const =0 |
Return strides of the input/output tensor of algorithm. | |
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().