|
| | Layer () |
| |
| | Layer (const LayerParams ¶ms) |
| | 仅初始化 名称、类型 和 数据块 字段。
|
| |
| virtual | ~Layer () |
| |
| virtual void | applyHalideScheduler (Ptr< BackendNode > &node, const std::vector< Mat * > &inputs, const std::vector< Mat > &outputs, int targetId) const |
| | 基于层超参数的自动 Halide 调度。
|
| |
| virtual void | finalize (const std::vector< Mat * > &input, std::vector< Mat > &output) |
| | 根据输入、输出和数据块计算和设置内部参数。
|
| |
| std::vector< Mat > | finalize (const std::vector< Mat > &inputs) |
| | 这是一个为了方便而提供的重载成员函数。它和上面的函数不同的地方在于它接受的参数。
|
| |
| void | finalize (const std::vector< Mat > &inputs, std::vector< Mat > &outputs) |
| | 这是一个为了方便而提供的重载成员函数。它和上面的函数不同的地方在于它接受的参数。
|
| |
| virtual void | finalize (InputArrayOfArrays inputs, OutputArrayOfArrays outputs) |
| | 根据输入、输出和数据块计算和设置内部参数。
|
| |
| virtual void | forward (InputArrayOfArrays inputs, OutputArrayOfArrays outputs, OutputArrayOfArrays internals) |
| | 给定输入的 blobs,计算输出 blobs。
|
| |
| virtual void | forward (std::vector< Mat * > &input, std::vector< Mat > &output, std::vector< Mat > &internals) |
| | 给定输入的 blobs,计算输出 blobs。
|
| |
| void | forward_fallback (InputArrayOfArrays inputs, OutputArrayOfArrays outputs, OutputArrayOfArrays internals) |
| | 给定输入的 blobs,计算输出 blobs。
|
| |
| virtual int64 | getFLOPS (const std::vector< MatShape > &inputs, const std::vector< MatShape > &outputs) const |
| |
| virtual bool | getMemoryShapes (const std::vector< MatShape > &inputs, const int requiredOutputs, std::vector< MatShape > &outputs, std::vector< MatShape > &internals) const |
| |
| virtual void | getScaleShift (Mat &scale, Mat &shift) const |
| | 返回具有通道乘法和加法的层的参数。
|
| |
| virtual void | getScaleZeropoint (float &scale, int &zeropoint) const |
| | 返回层的比例和零点。
|
| |
| virtual Ptr< BackendNode > | initCann (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendWrapper > > &outputs, const std::vector< Ptr< BackendNode > > &nodes) |
| | 返回一个 CANN 后端节点。
|
| |
| virtual Ptr< BackendNode > | initCUDA (void *context, const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendWrapper > > &outputs) |
| | 返回一个 CUDA 后端节点。
|
| |
| virtual Ptr< BackendNode > | initHalide (const std::vector< Ptr< BackendWrapper > > &inputs) |
| | 返回 Halide 后端节点。
|
| |
| virtual Ptr< BackendNode > | initNgraph (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendNode > > &nodes) |
| |
| virtual Ptr< BackendNode > | initTimVX (void *timVxInfo, const std::vector< Ptr< BackendWrapper > > &inputsWrapper, const std::vector< Ptr< BackendWrapper > > &outputsWrapper, bool isLast) |
| | 返回一个 TimVX 后端节点。
|
| |
| virtual Ptr< BackendNode > | initVkCom (const std::vector< < BackendWrapper > > &inputs, std::vector< Ptr< BackendWrapper > > &outputs) |
| |
| virtual Ptr< BackendNode > | initWebnn (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< >) &nodes) |
| |
| virtual int | inputNameToIndex (String inputName) |
| | 返回输入数组中输入块所在的索引。
|
| |
| virtual int | outputNameToIndex (const String &outputName) |
| | 返回输出数组中输出块所在的索引。
|
| |
| void | run (const std::vector< Mat > &inputs, std::vector< Mat > &outputs, std::vector< Mat > &internals) |
| | 分配层并计算输出。
|
| |
| virtual bool | setActivation (const ActivationLayer > &layer) |
| | 尝试将后续激活层附加到层上,即在部分情况下进行层融合。
|
| |
| void | | | 仅初始化 名称、类型 和 数据块 字段。
|
| |
| virtual bool | supportBackend (int backendId) |
| | 询问层是否支持特定的后端进行计算。
|
| |
| virtual Ptr< BackendNode > | eq id / < a宁 href=" href="../../dc/d84/group__core__basic.html#ga6395ca871a678020c4a31fadf7e8cc63">Ptr< >" class "el" (href../../discard href>>) > &node) |
| | 实现层的融合。
|
| |
| virtual bool | ] > &top)}} |
| | 尝试将与当前层相邻的层融合。
|
| |
| virtual bool | ';
|
| | 尝试量化特定的层,并计算固定点实现所需的量化参数。
|
| |
| virtual void | unsetAttached () |
| | 将所有附加到特定层的层“断开连接”。
|
| |
| virtual bool | updateMemoryShapes (const std::vector< MatShape > &inputs) |
| |
| | Algorithm () |
| |
| virtual | ~Algorithm () |
| |
| virtual void | clear () |
| | 清除算法状态。
|
| |
| virtual bool | empty () const |
| | 如果算法为空(例如在开始时或在读取失败后),则返回 true。
|
| |
| virtual String | getDefaultName () const |
| |
| virtual void | read (const FileNode &fn) |
| | 从文件存储中读取算法参数。
|
| |
| virtual void | save (const String &filename) const |
| |
| void | write (const Ptr< FileStorage > &fs, const String &name=String()) const |
| |
| virtual void | write (FileStorage &fs) const |
| | 将算法参数存储在文件存储中。
|
| |
| void | write (FileStorage &fs, const String &name) const |
| |
| std::vector< Mat > | blobs |
| | 必须存储学到的参数列表,以允许使用 Net::getParam() 读取它们。
|
| |
| String | name |
| | 层的实例名称,可用于日志或其他内部用途。
|
| |
| int | preferableTarget |
| | preference target for layer forwarding
|
| |
| String | type |
| | 用于通过层工厂创建层的类型名称。
|
| |
| void | writeFormat (FileStorage &fs) const |
| |