From 0a51abe8ab71c03eff66200265142fc910bcbfe2 Mon Sep 17 00:00:00 2001 From: "Guo, Yejun" Date: Fri, 7 Aug 2020 14:32:55 +0800 Subject: dnn: add backend options when load the model different backend might need different options for a better performance, so, add the parameter into dnn interface, as a preparation. Signed-off-by: Guo, Yejun --- libavfilter/dnn/dnn_backend_tf.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'libavfilter/dnn/dnn_backend_tf.c') diff --git a/libavfilter/dnn/dnn_backend_tf.c b/libavfilter/dnn/dnn_backend_tf.c index 9ceca5cea0..86da99160d 100644 --- a/libavfilter/dnn/dnn_backend_tf.c +++ b/libavfilter/dnn/dnn_backend_tf.c @@ -572,7 +572,7 @@ static DNNReturnType load_native_model(TFModel *tf_model, const char *model_file return DNN_SUCCESS; } -DNNModel *ff_dnn_load_model_tf(const char *model_filename) +DNNModel *ff_dnn_load_model_tf(const char *model_filename, const char *options) { DNNModel *model = NULL; TFModel *tf_model = NULL; @@ -600,6 +600,7 @@ DNNModel *ff_dnn_load_model_tf(const char *model_filename) model->model = (void *)tf_model; model->set_input_output = &set_input_output_tf; model->get_input = &get_input_tf; + model->options = options; return model; } -- cgit v1.2.3