|
| Layer () |
|
| Layer (const LayerParams ¶ms) |
| 仅初始化 名称、类型 和 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) |
| 根据输入、输出和 blobs 计算并设置内部参数。
|
|
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) |
| 根据输入、输出和 blobs 计算并设置内部参数。
|
|
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<Ptr<BackendWrapper>> &inputs, std::vector<Ptr<BackendWrapper>> &outputs) |
|
virtual Ptr< BackendNode > | initWebnn (const std::vector<Ptr<BackendWrapper>> &inputs, const std::vector<Ptr<BackendNode>> &nodes) |
|
虚拟 int | inputNameToIndex (String inputName) |
| 返回输入数组中输入 blob 的索引。
|
|
虚拟 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 Ptr<ActivationLayer>) |
| 尝试将后续的激活层附加到层上,即执行部分情况下的层融合。
|
|
void | setParamsFrom (const LayerParams ¶ms) |
| 仅初始化 名称、类型 和 blobs 字段。
|
|
virtual bool | supportBackend (int backendId) |
| 询问层是否支持特定后端进行计算。
|
|
virtual Ptr< BackendNode > | tryAttach (const Ptr<BackendNode>) |
| 实现层融合。
|
|
virtual bool | tryFuse (Ptr<Layer>) |
| 尝试将与当前层下一个层融合。
|
|
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 |
| prefer target for layer forwarding
|
|
String | type |
| 用于通过层工厂创建层的类型名。
|
|
void | writeFormat (FileStorage &fs) const |
|