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

Properly handle shape nodes that have a preexisting control dependency

Browse Source 
PiperOrigin-RevId: 158274845
This commit is contained in:
Benoit Steiner 2017-06-07 08:59:00 -07:00 committed by TensorFlower Gardener
parent 1d67379d56
commit 3b6fe94bbb
1 changed files with 6 additions and 4 deletions

10
tensorflow/core/grappler/optimizers/constant_folding.cc
View File

@ -229,11 +229,13 @@ Status ConstantFolding::MaterializeShapes(const GrapplerItem& item) {
value.AsProtoTensorContent(
(*node.mutable_attr())["value"].mutable_tensor());
// Turn the inputs into control dependencies: this is needed to ensure
// that the constant value will only be generated in the cases where the
// shape/rank/size would have been generated in the original graph.
// Turn the data input into a control dependency: this is needed to
// ensure that the constant value will only be generated in the cases
// where the shape/rank/size would have been generated in the original
// graph. Additional inputs are extra control dependencies that we
// preserve.
CHECK_LE(1, node.input_size());
string ctrl_dep = AddControlDependency(node.input(0));
CHECK_EQ(1, node.input_size());
node.set_input(0, ctrl_dep);
}
}