From c9abf7eae8823e7b32833b347ef0ddc64b1ffff7 Mon Sep 17 00:00:00 2001
From: xueteng <xueteng@huawei.com>
Date: Tue, 24 May 2022 10:05:02 +0800
Subject: [PATCH] ut

---
 parser/tensorflow/tensorflow_parser.cc | 3 +++
 1 file changed, 3 insertions(+)

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;
     }
   }