diff mbox

[FFmpeg-devel,1/8] libavfilter/dnn_backend_tf.c: set layer_add_res for input layer

Message ID 1554215315-13776-1-git-send-email-yejun.guo@intel.com
State Accepted
Commit 014b6a56f8e4020dd9ec2255266cd0a24c885c0c
Headers show

Commit Message

Guo, Yejun April 2, 2019, 2:28 p.m. UTC
otherwise, the following check will return error if layer_add_res
is randomly initialized.

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
---
 libavfilter/dnn_backend_tf.c | 1 +
 1 file changed, 1 insertion(+)

Comments

Liu Steven April 16, 2019, 11:55 a.m. UTC | #1
> 在 2019年4月2日,22:28,Guo, Yejun <yejun.guo@intel.com> 写道:
> 
> otherwise, the following check will return error if layer_add_res
> is randomly initialized.
> 
> Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
> ---
> libavfilter/dnn_backend_tf.c | 1 +
> 1 file changed, 1 insertion(+)
> 
> diff --git a/libavfilter/dnn_backend_tf.c b/libavfilter/dnn_backend_tf.c
> index 5bc7f06..9e0c127 100644
> --- a/libavfilter/dnn_backend_tf.c
> +++ b/libavfilter/dnn_backend_tf.c
> @@ -440,6 +440,7 @@ static DNNReturnType load_native_model(TFModel *tf_model, const char *model_file
>     for (layer = 0; layer < conv_network->layers_num; ++layer){
>         switch (conv_network->layers[layer].type){
>         case INPUT:
> +            layer_add_res = DNN_SUCCESS;
>             break;
>         case CONV:
>             layer_add_res = add_conv_layer(tf_model, transpose_op, &op,
> -- 
> 2.7.4
> 
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel@ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
> 
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe”.

LGTM

Thanks
Steven
diff mbox

Patch

diff --git a/libavfilter/dnn_backend_tf.c b/libavfilter/dnn_backend_tf.c
index 5bc7f06..9e0c127 100644
--- a/libavfilter/dnn_backend_tf.c
+++ b/libavfilter/dnn_backend_tf.c
@@ -440,6 +440,7 @@  static DNNReturnType load_native_model(TFModel *tf_model, const char *model_file
     for (layer = 0; layer < conv_network->layers_num; ++layer){
         switch (conv_network->layers[layer].type){
         case INPUT:
+            layer_add_res = DNN_SUCCESS;
             break;
         case CONV:
             layer_add_res = add_conv_layer(tf_model, transpose_op, &op,