TensorRT
7.1.3.0
|
Describes a variation of execution of a layer. An algorithm is represented by IAlgorithmVariant and the IAlgorithmIOInfo for each of its inputs and outputs. An algorithm can be selected or reproduced using AlgorithmSelector::selectAlgorithms().". More...
#include <NvInfer.h>
Public Member Functions | |
virtual const IAlgorithmIOInfo & | getAlgorithmIOInfo (int32_t index) const =0 |
Returns the format of an Algorithm input or output. Algorithm inputs are incrementally numbered first, followed by algorithm outputs. More... | |
virtual const IAlgorithmVariant & | getAlgorithmVariant () const =0 |
Returns the algorithm variant. | |
virtual float | getTimingMSec () const =0 |
The time in milliseconds to execute the algorithm. | |
virtual std::size_t | getWorkspaceSize () const =0 |
The size of the GPU temporary memory in bytes which the algorithm uses at execution time. | |
Describes a variation of execution of a layer. An algorithm is represented by IAlgorithmVariant and the IAlgorithmIOInfo for each of its inputs and outputs. An algorithm can be selected or reproduced using AlgorithmSelector::selectAlgorithms().".
|
pure virtual |
Returns the format of an Algorithm input or output. Algorithm inputs are incrementally numbered first, followed by algorithm outputs.
index | Index of the input or output of the algorithm. Incremental numbers assigned to indices of inputs and the outputs. |