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