TensorRT  7.1.3.0
nvinfer1::IAlgorithm Class Referenceabstract

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 IAlgorithmIOInfogetAlgorithmIOInfo (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 IAlgorithmVariantgetAlgorithmVariant () 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.
 

Detailed Description

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.

Member Function Documentation

◆ 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
indexIndex of the input or output of the algorithm. Incremental numbers assigned to indices of inputs and the outputs.

The documentation for this class was generated from the following file: