diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc
index b258a54..c6c57ed 100644
--- a/parser/tensorflow/tensorflow_parser.cc
+++ b/parser/tensorflow/tensorflow_parser.cc
@@ -2500,6 +2500,9 @@ Status TensorFlowModelParser::OptimizeIdentityByOutput(map<string, NodeDef *> &n
       output_node_def->set_input(0, curr_node_def->input(0).c_str());
       has_out_retval = true;
       GELOGD("op %s set input(0):%s.", output_node_def->name().c_str(), curr_node_def->input(0).c_str());
+    } else {
+      has_out_retval = false;
+      break;
     }
   }