diff --git a/lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc b/lite/core/mir/elimination/remove_tf_redundant_ops_pass.cc index 792d27b4292e1661ae3ec07ec2a4f649c9361e63..9defc2dacf4d50ad5a1d2ed7aa5e3afcb16cabb9 100644 --- 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()