#include <NvInfer.h>

Inheritance diagram for nvinfer1::v_1_0::IAlgorithmSelector:

Public Member Functions
InterfaceInfo	getInterfaceInfo () const noexcept override
	Return version information associated with this interface. Applications must not override this method. More...

virtual int32_t	selectAlgorithms (IAlgorithmContext const &context, IAlgorithm const const choices, int32_t nbChoices, int32_t *selection) noexcept=0
	Select Algorithms for a layer from the given list of algorithm choices. More...

virtual void	reportAlgorithms (IAlgorithmContext const const algoContexts, IAlgorithm const const algoChoices, int32_t nbAlgorithms) noexcept=0
	Called by TensorRT to report choices it made. More...

virtual	~IAlgorithmSelector () noexcept=default

Public Member Functions inherited from nvinfer1::IVersionedInterface
virtual APILanguage	getAPILanguage () const noexcept
	The language used to build the implementation of this Interface. More...

virtual	~IVersionedInterface () noexcept=default

Additional Inherited Members
Protected Member Functions inherited from nvinfer1::IVersionedInterface
	IVersionedInterface ()=default

	IVersionedInterface (IVersionedInterface const &)=default

	IVersionedInterface (IVersionedInterface &&)=default

IVersionedInterface &	operator= (IVersionedInterface const &) &=default

IVersionedInterface &	operator= (IVersionedInterface &&) &=default

Constructor & Destructor Documentation

◆ ~IAlgorithmSelector()

virtual nvinfer1::v_1_0::IAlgorithmSelector::~IAlgorithmSelector ( )

virtualdefaultnoexcept

InterfaceInfo nvinfer1::v_1_0::IAlgorithmSelector::getInterfaceInfo ( ) const

inlineoverridevirtualnoexcept

Return version information associated with this interface. Applications must not override this method.

virtual void nvinfer1::v_1_0::IAlgorithmSelector::reportAlgorithms	(	IAlgorithmContext const const	algoContexts,
		IAlgorithm const const	algoChoices,
		int32_t	nbAlgorithms
	)

pure virtualnoexcept

Called by TensorRT to report choices it made.

Note: For a given optimization profile, this call comes after all calls to selectAlgorithms. algoChoices[i] is the choice that TensorRT made for algoContexts[i], for i in [0, nbAlgorithms-1]

Parameters

algoContexts	The list of all algorithm contexts.
algoChoices	The list of algorithm choices made by TensorRT
nbAlgorithms	The size of algoContexts as well as algoChoices.

virtual int32_t nvinfer1::v_1_0::IAlgorithmSelector::selectAlgorithms	(	IAlgorithmContext const &	context,
		IAlgorithm const const	choices,
		int32_t	nbChoices,
		int32_t *	selection
	)

pure virtualnoexcept

Select Algorithms for a layer from the given list of algorithm choices.

Parameters

context	The context for which the algorithm choices are valid.
choices	The list of algorithm choices to select for implementation of this layer.
nbChoices	Number of algorithm choices.
selection	The user writes indices of selected choices in to selection buffer which is of size nbChoices.

Note: TensorRT uses its default algorithm selection to choose from the list provided. If return value is 0, TensorRT's default algorithm selection is used unless BuilderFlag::kREJECT_EMPTY_ALGORITHMS is set. The list of choices is valid only for this specific algorithm context.

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