TensorRT  7.1.3.0
nvinfer1::IAlgorithmContext Class Referenceabstract

Describes the context and requirements, that could be fulfilled by one or more instances of IAlgorithm. More...

#include <NvInfer.h>

Public Member Functions

virtual const char * getName () const =0
 Return name of the algorithm node. This is a unique identifier for the IAlgorithmContext.
 
virtual Dims getDimensions (int32_t index, OptProfileSelector select) const =0
 Get the minimum / optimum / maximum dimensions for input or output tensor. More...
 
virtual int32_t getNbInputs () const =0
 Return number of inputs of the algorithm.
 
virtual int32_t getNbOutputs () const =0
 Return number of outputs of the algorithm.
 

Detailed Description

Describes the context and requirements, that could be fulfilled by one or more instances of IAlgorithm.

See also
IAlgorithm
Warning
Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

◆ getDimensions()

virtual Dims nvinfer1::IAlgorithmContext::getDimensions ( int32_t  index,
OptProfileSelector  select 
) const
pure virtual

Get the minimum / optimum / maximum dimensions for input or output tensor.

Parameters
indexIndex of the input or output of the algorithm. Incremental numbers assigned to indices of inputs and the outputs.
selectWhich of the minimum, optimum, or maximum dimensions to be queried.

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