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

Merge pull request #15366 from evision-ai/fix-remove-nodes

Browse Source 
fix that remove_nodes drops input suffixes
This commit is contained in:
Martin Wicke 2018-02-15 16:31:55 -08:00 committed by GitHub
commit 57b583369d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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);