nvinfer1::IAlgorithm 类参考

描述了层的执行变体。算法由 IAlgorithmVariant 和每个输入和输出的 IAlgorithmIOInfo 表示。可以使用 AlgorithmSelector::selectAlgorithms() 选择或重现算法。 更多...

#include <NvInfer.h>

nvinfer1::IAlgorithm 的继承关系图

公共成员函数
IAlgorithmVariant const &	getAlgorithmVariant () const noexcept
	返回算法变体。 更多...
float	getTimingMSec () const noexcept
	执行算法的时间，以毫秒为单位。 更多...
std::size_t	getWorkspaceSize () const noexcept
	算法在执行时使用的 GPU 临时内存大小，以字节为单位。 更多...
IAlgorithmIOInfo const *	getAlgorithmIOInfoByIndex (int32_t index) const noexcept
	返回算法输入或输出的格式。算法输入首先按递增顺序编号，然后是算法输出。 更多...

受保护的成员函数
virtual	~IAlgorithm () noexcept=default
继承自 nvinfer1::INoCopy 的受保护的成员函数
	INoCopy ()=default
virtual	~INoCopy ()=default
	INoCopy (INoCopy const &other)=delete
INoCopy &	operator= (INoCopy const &other)=delete
	INoCopy (INoCopy &&other)=delete
INoCopy &	operator= (INoCopy &&other)=delete

受保护的属性
apiv::VAlgorithm *	mImpl

详细描述

描述了层的执行变体。算法由 IAlgorithmVariant 和每个输入和输出的 IAlgorithmIOInfo 表示。可以使用 AlgorithmSelector::selectAlgorithms() 选择或重现算法。

参见

IAlgorithmIOInfo, IAlgorithmVariant, IAlgorithmSelector::selectAlgorithms()

警告

请勿从此类继承，因为这样做会破坏 API 和 ABI 的向前兼容性。

已弃用

在 TensorRT 10.8 中已弃用。请在 ITimingCache 中使用可编辑模式代替。

构造函数 & 析构函数文档

~IAlgorithm()

virtual nvinfer1::IAlgorithm::~IAlgorithm ( )

protectedvirtualdefaultnoexcept

成员函数文档

getAlgorithmIOInfoByIndex()

IAlgorithmIOInfo const * nvinfer1::IAlgorithm::getAlgorithmIOInfoByIndex ( int32_t  index ) const

inlinenoexcept

返回算法输入或输出的格式。算法输入首先按递增顺序编号，然后是算法输出。

参数

index

算法的输入或输出的索引。分配给输入和输出索引的递增数字。

返回

指向 IAlgorithmIOInfo 接口的指针，如果索引超出范围，则为 nullptr。

getAlgorithmVariant()

IAlgorithmVariant const & nvinfer1::IAlgorithm::getAlgorithmVariant ( ) const

inlinenoexcept

返回算法变体。

getTimingMSec()

float nvinfer1::IAlgorithm::getTimingMSec ( ) const

inlinenoexcept

执行算法的时间，以毫秒为单位。

getWorkspaceSize()

std::size_t nvinfer1::IAlgorithm::getWorkspaceSize ( ) const

inlinenoexcept

算法在执行时使用的 GPU 临时内存大小，以字节为单位。

成员数据文档

mImpl

apiv::VAlgorithm* nvinfer1::IAlgorithm::mImpl

protected

---

此类文档从以下文件生成

• NvInfer.h