From 3808773f9a88efad664b9514d350cb4c77d6a978 Mon Sep 17 00:00:00 2001 From: xueteng Date: Thu, 2 Jun 2022 10:46:50 +0800 Subject: [PATCH] log --- parser/tensorflow/tensorflow_parser.cc | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index b258a54..6a67949 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -2496,10 +2496,14 @@ Status TensorFlowModelParser::OptimizeIdentityByOutput(map &n domi::tensorflow::NodeDef *output_node_def = nodedef_map[output_node_name]; GE_CHECK_NOTNULL(output_node_def); if (output_node_def->op() == "_Retval") { - GELOGD("_Retval Identity need optimize."); + GELOGW("_Retval Identity need optimize. node:%s", curr_node_name.c_str()); 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()); + GELOGW("op %s set input(0):%s.", output_node_def->name().c_str(), curr_node_def->input(0).c_str()); + } else { + GELOGW("[test]current node(%s).", curr_node_name.c_str()); + has_out_retval = false; + break; } }