Strip the :x suffix when generating control inputs from input names

PiperOrigin-RevId: 157770257
This commit is contained in:
Benoit Steiner 2017-06-01 15:17:54 -07:00 committed by TensorFlower Gardener
parent d6fe47af57
commit cc346e6906

View File

@ -176,7 +176,7 @@ Status ConstantFolding::MaterializeShapes(const GrapplerItem& item) {
// Turn the inputs into control dependencies.
CHECK_EQ(1, node.input_size());
node.set_input(0, strings::StrCat("^", node.input(0)));
node.set_input(0, strings::StrCat("^", NodeName(node.input(0))));
}
}
}