|
| 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) |
| | 根據輸入、輸出和資料塊計算並設定內部引數。
|
| |
| 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) |
| | 給定 input 資料塊,計算輸出 blobs。
|
| |
| virtual void | forward (std::vector< Mat * > &input, std::vector< Mat > &output, std::vector< Mat > &internals) |
| | 給定 input 資料塊,計算輸出 blobs。
|
| |
| void | forward_fallback (InputArrayOfArrays inputs, OutputArrayOfArrays outputs, OutputArrayOfArrays internals) |
| | 給定 input 資料塊,計算輸出 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) |
| |
| 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 Ptr< ActivationLayer > &layer) |
| | 嘗試將後續啟用層附加到當前層,即在部分情況下進行層融合。
|
| |
| 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 |
| |
| void | writeFormat (FileStorage &fs) const |
| |