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

In fusion also rename explicitly parameters like input_data_X_w to the new indices.

Browse Source 
PiperOrigin-RevId: 243659452
This commit is contained in:
A. Unique TensorFlower 2019-04-15 11:52:11 -07:00 committed by TensorFlower Gardener
parent 72dd4cf622
commit 6763405490
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/delegates/gpu/gl/compiler/fuse_auto_input.cc
View File

@ -180,6 +180,11 @@ TransformResult FuseAutoInput::ApplyToNode(Node* node, GraphFloat32* graph) {
}
input_num++;
}
// Also rename all _h and _w parameters to the new names.
for (auto& param : attr.code.parameters) {
param.name = absl::StrReplaceAll(param.name, replacements);
}
attr.code.source_code =
absl::StrReplaceAll(attr.code.source_code, replacements);