|
virtual void | forwardSlice (const float *src, float *dst, int len, size_t outPlaneSize, int cn0, int cn1) const |
|
virtual void | forwardSlice (const int *src, const int *lut, int *dst, int len, size_t outPlaneSize, int cn0, int cn1) const |
|
virtual void | forwardSlice (const int8_t *src, const int8_t *lut, int8_t *dst, int len, size_t outPlaneSize, int cn0, int cn1) const |
|
| 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 (const std::vector< Mat > &inputs, std::vector< Mat > &outputs) |
| 这是一个重载的成员函数,它只是为了方便。它与上述函数不同之处仅仅在于它接受的参数。
|
|
virtual void | finalize (InputArrayOfArrays inputs, OutputArrayOfArrays outputs) |
| 根据输入、输出和 blob 计算和设置内部参数。
|
|
virtual void | forward (InputArrayOfArrays inputs, OutputArrayOfArrays outputs, OutputArrayOfArrays internals) |
| 指定 input blob 后,计算输出 blobs 。
|
|
virtual void | forward(std::vector< Mat * > &输入, std::vector< Mat > &输出, std::vector< Mat > &内部) |
| 指定 input blob 后,计算输出 blobs 。
|
|
void | forward_fallback(InputArrayOfArrays 输入, OutputArrayOfArrays 输出, OutputArrayOfArrays 内部) |
| 指定 input blob 后,计算输出 blobs 。
|
|
virtual int64 | getFLOPS(const std::vector< MatShape > &输入, const std::vector< MatShape > &输出)const |
|
virtual bool | getMemoryShapes(const std::vector< MatShape > &输入, const int requiredOutputs, std::vector< MatShape > &输出, std::vector< MatShape > &内部)const |
|
virtual void | getScaleShift(Mat &缩放, Mat &位移)const |
| 返回具有逐通道乘法和加法的层的参数。
|
|
virtual void | getScaleZeropoint(float &scale,int &zeropoint)const |
| 返回层的缩放和零点。
|
|
virtual Ptr< BackendNode > | initCann(const std::vector< Ptr< BackendWrapper > > &输入, const std::vector< Ptr< BackendWrapper > > &输出, const std::vector< Ptr< BackendNode > > &节点) |
| 返回一个 CANN 后端节点。
|
|
virtual Ptr< BackendNode > | initCUDA(void *context, const std::vector< Ptr< BackendWrapper > > |
| 返回 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< Ptr< BackendWrapper > > &inputs, std::vector< Ptr< BackendWrapper > > &outputs) |
|
virtual Ptr< BackendNode > | initWebnn (const std::vector< Ptr< BackendWrapper > > &inputs, const std::vector< Ptr< BackendNode > > &nodes) |
|
virtual int | inputNameToIndex (String inputName) |
| 返回输入数组内输入 blob 的索引。
|
|
virtual int | outputNameToIndex (const String &outputName) |
| 返回输出数组内输出 blob 的索引。
|
|
void | run (const std::vector< Mat > &inputs, std::vector< Mat > &outputs, std::vector< Mat > &internals) |
| 分配层并计算输出。
|
|
virtual bool | setActivation (const | | 尝试将后续激活层附加到层,即部分情况下进行层融合。
|
|
void | setParamsFrom (const LayerParams ¶ms) |
| 仅初始化 name、type 和 blobs 字段。
|
|
virtual bool | supportBackend (int backendId) |
| 询问层是否支持特定后端以进行计算。
|
|
virtual Ptr< BackendNode > | tryAttach (const Ptr< BackendNode > &node) |
| 实施层融合。
|
|
virtual bool | tryFuse (Ptr< Layer > &top) |
| 尝试将当前层与下一层融合。
|
|
virtual bool | tryQuantize (const std::vector< std::vector< float > > &scales, const std::vector< std::vector< int > > &zeropoints, LayerParams ¶ms) |
| 尝试对给定层进行量化并计算定点实现所需的量化参数。
|
|
virtual void | unsetAttached () |
| “分离”附加到特定层的全部层。
|
|
virtual bool | updateMemoryShapes (const std::vector< MatShape > &inputs) |
|
| Algorithm () |
|
virtual | ~Algorithm () |
|
virtual void | clear () |
| 清除算法状态。
|
|
virtual bool | empty () const |
| 如果 Algorithm 为空(例如,在开始阶段或读取不成功之后),则返回 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 |
| 图层前向传输的首选目标
|
|
String | type |
| 用于通过层工厂创建层的类型名称。
|
|
void | writeFormat (FileStorage &fs) const |
|