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...
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().".
- See also
- IAlgorithmIOInfo, IAlgorithmVariant, IAlgorithmSelector::selectAlgorithms()
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
◆ ~IAlgorithm()
virtual nvinfer1::IAlgorithm::~IAlgorithm |
( |
| ) |
|
|
inlineprotectedvirtual |
◆ getAlgorithmIOInfo()
virtual const IAlgorithmIOInfo& nvinfer1::IAlgorithm::getAlgorithmIOInfo |
( |
int32_t |
index | ) |
const |
|
pure virtual |
Returns the format of an Algorithm input or output.
Algorithm inputs are incrementally numbered first, followed by algorithm outputs.
- Parameters
-
index | Index of the input or output of the algorithm. Incremental numbers assigned to indices of inputs and the outputs. |
◆ getAlgorithmVariant()
Returns the algorithm variant.
◆ getTimingMSec()
virtual float nvinfer1::IAlgorithm::getTimingMSec |
( |
| ) |
const |
|
pure virtual |
The time in milliseconds to execute the algorithm.
◆ getWorkspaceSize()
virtual std::size_t nvinfer1::IAlgorithm::getWorkspaceSize |
( |
| ) |
const |
|
pure virtual |
The size of the GPU temporary memory in bytes which the algorithm uses at execution time.
The documentation for this class was generated from the following file: