TensorRT 10.8.0
|
表示行程计数限制器的层。 更多...
#include <NvInfer.h>
公共成员函数 | |
TripLimit | getTripLimit () const noexcept |
获取行程限制器类型。 更多... | |
![]() | |
ILoop * | getLoop () const noexcept |
获取与此边界层关联的 ILoop 的指针。 更多... | |
![]() | |
LayerType | getType () const noexcept |
返回层的类型。 更多... | |
void | setName (char const *name) noexcept |
设置层的名称。 更多... | |
char const * | getName () const noexcept |
返回层的名称。 更多... | |
int32_t | getNbInputs () const noexcept |
获取层的输入数量。 更多... | |
ITensor * | getInput (int32_t index) const noexcept |
获取与给定索引对应的层输入。 更多... | |
int32_t | getNbOutputs () const noexcept |
获取层的输出数量。 更多... | |
ITensor * | getOutput (int32_t index) const noexcept |
获取与给定索引对应的层输出。 更多... | |
void | setInput (int32_t index, ITensor &tensor) noexcept |
用特定的张量替换此层的输入。 更多... | |
void | setPrecision (DataType dataType) noexcept |
在弱类型网络中设置此层的首选或必需的计算精度。 更多... | |
DataType | getPrecision () const noexcept |
获取此层的计算精度 更多... | |
bool | precisionIsSet () const noexcept |
是否已为此层设置计算精度 更多... | |
void | resetPrecision () noexcept |
重置此层的计算精度 更多... | |
void | setOutputType (int32_t index, DataType dataType) noexcept |
在弱类型网络中设置此层的输出类型。 更多... | |
DataType | getOutputType (int32_t index) const noexcept |
获取此层的输出类型 更多... | |
bool | outputTypeIsSet (int32_t index) const noexcept |
是否已为此层设置输出类型 更多... | |
void | resetOutputType (int32_t index) noexcept |
重置此层的输出类型 更多... | |
void | setMetadata (char const *metadata) noexcept |
为此层设置元数据。 更多... | |
char const * | getMetadata () const noexcept |
获取层的元数据。 更多... | |
保护成员函数 | |
virtual | ~ITripLimitLayer () noexcept=default |
![]() | |
virtual | ~ILoopBoundaryLayer () noexcept=default |
![]() | |
virtual | ~ILayer () noexcept=default |
![]() | |
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::VTripLimitLayer * | mImpl |
![]() | |
apiv::VLoopBoundaryLayer * | mBoundary |
![]() | |
apiv::VLayer * | mLayer |
表示行程计数限制器的层。
行程限制层使用 kCOUNT 定义迭代次数,或使用 kWHILE 定义条件循环,从而设置循环的执行条件。一个循环可以具有每种限制中的一种,在这种情况下,当达到行程计数或条件变为假时,循环退出。
参见 INetworkDefinition::addTripLimit()。
|
protectedvirtualdefaultnoexcept |
|
inlinenoexcept |
获取行程限制器类型。
|
protected |
Copyright © 2024 NVIDIA Corporation
隐私政策 | 管理我的隐私 | 请勿出售或分享我的数据 | 服务条款 | 辅助功能 | 公司政策 | 产品安全 | 联系我们