diff options
| author | Guo, Yejun <yejun.guo@intel.com> | 2019-10-09 22:08:18 +0800 |
|---|---|---|
| committer | Pedro Arthur <bygrandao@gmail.com> | 2019-10-15 18:56:54 -0300 |
| commit | 2558e62713ebc5f3ea22c1a28d8e9cf3249badaf (patch) | |
| tree | c8b4dfe51ec30faa41f7327f15000d49c3f05c72 /libavfilter/dnn/dnn_backend_native_layers.h | |
| parent | 3fd5ac7e92049b4f31026acdb53a762289f71448 (diff) | |
avfilter/dnn: unify the layer load function in native mode
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native_layers.h')
| -rw-r--r-- | libavfilter/dnn/dnn_backend_native_layers.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layers.h b/libavfilter/dnn/dnn_backend_native_layers.h index 3276aeea92..2df0ce9953 100644 --- a/libavfilter/dnn/dnn_backend_native_layers.h +++ b/libavfilter/dnn/dnn_backend_native_layers.h @@ -26,7 +26,13 @@ typedef int (*LAYER_EXEC_FUNC)(DnnOperand *operands, const int32_t *input_operand_indexes, int32_t output_operand_index, const void *parameters); +typedef int (*LAYER_LOAD_FUNC)(Layer *layer, AVIOContext *model_file_context, int file_size); -extern LAYER_EXEC_FUNC layer_funcs[DLT_COUNT]; +typedef struct LayerFunc { + LAYER_EXEC_FUNC pf_exec; + LAYER_LOAD_FUNC pf_load; +}LayerFunc; + +extern LayerFunc layer_funcs[DLT_COUNT]; #endif |