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