Browse Source

ut

pull/562/head
xueteng 3 years ago
parent
commit
8e59b24abb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      parser/tensorflow/tensorflow_parser.cc

+ 1
- 1
parser/tensorflow/tensorflow_parser.cc View File

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


Loading…
Cancel
Save