experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix that remove_nodes drops input suffixes

Browse Source
This commit is contained in:
gdh1995 2017-12-14 20:26:54 +08:00
parent 18c864cdae
commit 81c1568b3d
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/tools/graph_transforms/remove_nodes.cc
View File

@ -81,7 +81,17 @@ Status RemoveNodes(const GraphDef& input_graph_def,
return Status::OK(); return Status::OK();
} }
const NodeDef& input_node = match.inputs[0].node; const NodeDef& input_node = match.inputs[0].node;
inputs_to_rename[replace_node.name()] = input_node.name(); string target_name = input_node.name();
for (const string& input : replace_node.input()) {
if (!input.compare(0, target_name.size(), target_name)) {
if (input.size() == target_name.size() ||
input[target_name.size()] == ':') {
target_name = input;
break;
}
}
}
inputs_to_rename[replace_node.name()] = target_name;
inputs_to_rename["^" + replace_node.name()] = inputs_to_rename["^" + replace_node.name()] =
"^" + input_node.name(); "^" + input_node.name();
new_nodes->push_back(input_node); new_nodes->push_back(input_node);