summaryrefslogtreecommitdiff
path: root/libavfilter/dnn/dnn_backend_native_layer_mathunary.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_layer_mathunary.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_layer_mathunary.h')
-rw-r--r--libavfilter/dnn/dnn_backend_native_layer_mathunary.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathunary.h b/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
index 0467717a8b..40a9bb5fb8 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
+++ b/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
@@ -41,7 +41,7 @@ typedef struct DnnLayerMathUnaryParams{
DNNMathUnaryOperation un_op;
} DnnLayerMathUnaryParams;
-int dnn_load_layer_math_unary(Layer *layer, AVIOContext *model_file_context, int file_size);
+int dnn_load_layer_math_unary(Layer *layer, AVIOContext *model_file_context, int file_size, int operands_num);
int dnn_execute_layer_math_unary(DnnOperand *operands, const int32_t *input_operand_indexes,
int32_t output_operand_index, const void *parameters);