MinimumWorkspaceAlgorithmSelector Class Reference

Selects Algorithms with minimum workspace requirements. More...

Inheritance diagram for MinimumWorkspaceAlgorithmSelector:

Collaboration diagram for MinimumWorkspaceAlgorithmSelector:

Public Member Functions
int32_t	selectAlgorithms (const nvinfer1::IAlgorithmContext &algoContext, const nvinfer1::IAlgorithm *const *algoChoices, int32_t nbChoices, int32_t *selection) override
	Return value in [0, nbChoices] for a valid algorithm. More...
void	reportAlgorithms (const nvinfer1::IAlgorithmContext *const *algoContexts, const nvinfer1::IAlgorithm *const *algoChoices, int32_t nbAlgorithms) override
	Called by TensorRT to report choices it made. More...

Detailed Description

Selects Algorithms with minimum workspace requirements.

Member Function Documentation

selectAlgorithms()

int32_t MinimumWorkspaceAlgorithmSelector::selectAlgorithms ( const nvinfer1::IAlgorithmContext &  algoContext, const nvinfer1::IAlgorithm *const *  algoChoices, int32_t  nbChoices, int32_t *  selection  )

inlineoverridevirtual

Return value in [0, nbChoices] for a valid algorithm.

Use the map created from cache to select algorithms.

Implements nvinfer1::IAlgorithmSelector.

reportAlgorithms()

void MinimumWorkspaceAlgorithmSelector::reportAlgorithms ( const nvinfer1::IAlgorithmContext *const *  algoContexts, const nvinfer1::IAlgorithm *const *  algoChoices, int32_t  nbAlgorithms  )

inlineoverridevirtual

Called by TensorRT to report choices it made.

Implements nvinfer1::IAlgorithmSelector.

---

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

• sampleAlgorithmSelector.cpp