|
|
|
@@ -2520,7 +2520,7 @@ Status TensorFlowModelParser::OptimizeIdentityByOutput(map<string, NodeDef *> &n |
|
|
|
GE_RETURN_IF_ERROR(CheckInputNodeName(output_node_def->input(k), &node_name, nullptr, &is_control)); |
|
|
|
|
|
|
|
GE_IF_BOOL_EXEC( |
|
|
|
node_name == curr_node_name, output_node_def->set_input(k, is_control ? ("^" + curr_node_def->input(0).c_str()) : curr_node_def->input(0).c_str()); |
|
|
|
node_name == curr_node_name, output_node_def->set_input(k, is_control ? ("^" + curr_node_def->input(0)).c_str() : curr_node_def->input(0).c_str()); |
|
|
|
GELOGD("%s op set input(%d):%s, is_control:%d.", output_node_def->name().c_str(), k, curr_node_def->input(0).c_str(), is_control);) |
|
|
|
} |
|
|
|
} |
|
|
|
|