Browse Source

log

pull/544/head
xueteng 3 years ago
parent
commit
3808773f9a
1 changed files with 6 additions and 2 deletions
  1. +6
    -2
      parser/tensorflow/tensorflow_parser.cc

+ 6
- 2
parser/tensorflow/tensorflow_parser.cc View File

@@ -2496,10 +2496,14 @@ Status TensorFlowModelParser::OptimizeIdentityByOutput(map<string, NodeDef *> &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;
}
}



Loading…
Cancel
Save