fix remove tf redundant op pass. test=develop (#4158)

1ebbfd97 · ysh329 · GitHub · 0ca19802 · 1ebbfd97
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc +5 -0

未找到文件。
--- a/lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc
+++ b/lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc
@@ -184,6 +184,11 @@ void RemoveTFRedundantOpsPass::RemoveSqueeze2Reshape2Pattern(
      }
    }

+    if (nullptr == reshape2_out_node) {
+      VLOG(5) << "reshape2_out_node doesn't found, skip now";
+      return;
+    }
+
    // find next inst node of reshape2
    VLOG(5) << "reshape2_out_node->outlinks.size():"
            << reshape2_out_node->outlinks.size()