summaryrefslogtreecommitdiff
path: root/libavfilter/dnn/dnn_backend_native_layers.h
diff options
context:
space:
mode:
authorGuo Yejun <yejun.guo@intel.com>2020-06-10 13:36:11 +0800
committerGuo, Yejun <yejun.guo@intel.com>2020-06-17 13:42:52 +0800
commit0b3bd001ac1745d9d008a2d195817df57d7d1d14 (patch)
tree00fe52c18ac22476df43f24d533c6aef8f36965d /libavfilter/dnn/dnn_backend_native_layers.h
parentfc932195ab0c9c00fa0cd9620c60763d978d495b (diff)
dnn_backend_native: check operand index
it fixed the issue in https://trac.ffmpeg.org/ticket/8716
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native_layers.h')
-rw-r--r--libavfilter/dnn/dnn_backend_native_layers.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layers.h b/libavfilter/dnn/dnn_backend_native_layers.h
index 2df0ce9953..b696e9c6fa 100644
--- a/libavfilter/dnn/dnn_backend_native_layers.h
+++ b/libavfilter/dnn/dnn_backend_native_layers.h
@@ -26,7 +26,7 @@
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);
+typedef int (*LAYER_LOAD_FUNC)(Layer *layer, AVIOContext *model_file_context, int file_size, int operands_num);
typedef struct LayerFunc {
LAYER_EXEC_FUNC pf_exec;