这是循环边界层的基类。 更多...
#include <NvInfer.h>
nvinfer1::ILoopBoundaryLayer 的继承关系图
公共成员函数 | |
| ILoop * | getLoop () const noexcept |
| 获取指向与此边界层关联的 ILoop 的指针。 更多... | |
 继承自 nvinfer1::ILayer 的公共成员函数 | |
| 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 | ~ILoopBoundaryLayer () noexcept=default |
| 继承自 nvinfer1::ILayer 的保护成员函数 | |
| virtual | ~ILayer () 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::VLoopBoundaryLayer * | mBoundary |
| 继承自 nvinfer1::ILayer 的保护属性 | |
| apiv::VLayer * | mLayer |
详细描述
这是循环边界层的基类。
循环边界层用于在网络中定义循环，从而实现循环的实现。循环的边界层由类 ILoop 创建。
共有四种边界层。
- ITripLimitLayer：控制循环迭代的次数。
- IIterationLayer：迭代输入张量。
- IRecurrenceLayer：返回初始值或来自先前循环迭代的值。
- ILoopOutputLayer：从循环迭代生成输出张量。
构造函数 & 析构函数文档
◆ ~ILoopBoundaryLayer()
|
protectedvirtualdefaultnoexcept |
成员函数文档
◆ getLoop()
成员数据文档
◆ mBoundary
|
protected |
此类文档由以下文件生成
