|
| | Layer () |
| |
| | Layer (const LayerParams ¶ms) |
| | 仅初始化 name、type 和 blobs 字段。
|
| |
| 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) |
| | 根据输入、输出和 blob 计算并设置内部参数。
|
| |
| std::vector< Mat > | finalize (const std::vector< Mat > &inputs) |
| | 这是一个重载的成员函数,为了方便而提供。它与上述函数有所不同,仅在于它所接受的参数。
|
| |
| void | finalize (常量 std::vector< 垫 > &inputs、std::vector< 垫 > &outputs) |
| | 这是一个重载的成员函数,为了方便而提供。它与上述函数有所不同,仅在于它所接受的参数。
|
| |
| virtual void | finalize (输入数据条数数组输入、输出数据条数数组输出) |
| | 根据输入、输出和 blob 计算并设置内部参数。
|
| |
| virtual void | forward (输入数据条数数组输入、输出数据条数数组输出、输出数据条数数组内部) |
| | 给出 input 块,计算输出 块。
|
| |
| virtual void | forward (std::vector< 垫 * > &input、std::vector< 垫 > &output、std::vector< 垫 > &internals) |
| | 给出 input 块,计算输出 块。
|
| |
| void | forward_fallback (输入数据条数数组输入、输出数据条数数组输出、输出数据条数数组内部) |
| | 给出 input 块,计算输出 块。
|
| |
| virtual int64 | getFLOPS (const std::vector< 垫形状 > &inputs、const std::vector< 垫形状 > &outputs) const |
| |
| virtual bool | getMemoryShapes (const std::vector< 垫形状 > &inputs、const int requiredOutputs、std::vector< 垫形状 > &outputs、std::vector< 垫形状 > &internals) const |
| |
| virtual void | getScaleShift (垫 &放大、垫 &位移)const |
| | 返回带有通道逐位乘法和加法的层的参数。
|
| |
| virtual void | getScaleZeropoint (float &放大、int &零点)const |
| | 返回层的放大和零点。
|
| |
| virtual initCann (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendWrapper > > &outputs, const std::vector< Ptr< BackendNode > > &nodes) |
| | 返回一个 CANN 后端节点。
|
| |
| virtual initCUDA (void *context, const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendWrapper > > &outputs) |
| | 返回一个 CUDA 后端节点。
|
| |
| virtual initHalide (const std::vector< Ptr< BackendWrapper > > &inputs) |
| | 返回 Halide 后端节点。
|
| |
| virtual initNgraph (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendNode > > &nodes) |
| |
| virtual initTimVX (void *timVxInfo, const std::vector< Ptr< BackendWrapper > > &inputsWrapper, const std::vector< Ptr< BackendWrapper > > &outputsWrapper, bool isLast) |
| | 返回一个 TimVX 后端节点。
|
| |
| virtual initVkCom (const std::vector< Ptr< BackendWrapper > > &inputs, std::vector< Ptr< BackendWrapper > > &outputs) |
| |
| virtual initWebnn (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< |
| |
| 虚拟 int | inputNameToIndex (字符串 inputName) |
| | 返回输入数组中输入 Blob 的索引。
|
| |
| 虚拟 int | outputNameToIndex(常量 字符串 &outputName) |
| | 返回输出数组中输出 Blob 的索引。
|
| |
| void | 运行(const std::vector< 垫子 > &inputs, std::vector< 垫子 > &outputs, std::vector< 垫子 > &internals) |
| | 分配层并计算输出。
|
| |
| virtual bool | setActivation(const 指针< ActivationLayer > &layer) |
| | 尝试将后续激活层连接到层,即在部分情况下执行层融合。
|
| |
| void | setParamsFrom(const LayerParams ¶ms) |
| | 仅初始化 name、type 和 blobs 字段。
|
| |
| virtual bool | 支持后端(int backendId) |
| | 询问层是否支持特定后端进行计算。
|
| |
| virtual 尝试连接(const 指针< BackendNode > &node) |
| | 实现层融合。
|
| |
| virtual bool | 尝试融合(指针< 层 > &top) |
| | 尝试将当前层与下一层融合。
|
| |
| virtual bool | 尝试量化(const std::vector< std::vector< 浮点数 > > &scales, const std::vector< std::vector< int > > &zeropoints, LayerParams ¶ms) |
| | 尝试量化给定层并计算定点数实现所需的量化参数。
|
| |
| virtual void | unsetAttached () |
| | “分离”连接到特定层的全部层。
|
| |
| virtual bool | updateMemoryShapes(const std::vector< MatShape > &inputs) |
| |
| | 算法 () |
| |
| virtual | ~算法 () |
| |
| virtual void | 清除 () |
| | 清除算法状态。
|
| |
| virtual bool | empty()常量 |
| | 如果算法为空(例如在最开始或读取不成功的之后),则返回真。
|
| |
| 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 |
| |
| void | writeFormat (FileStorage &fs) const |
| |