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

Do not skip pruning identities inserted by Grappler in dependency optimizer.

Browse Source 
PiperOrigin-RevId: 236370572
This commit is contained in:
A. Unique TensorFlower 2019-03-01 13:44:12 -08:00 committed by TensorFlower Gardener
parent 0afe41679f
commit 8fa58e7762
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -75,16 +75,8 @@ bool DependencyOptimizer::SafeToRemoveIdentity(const NodeDef& node) const {
// Recv. // Recv.
if (IsVariable(*input) || IsRecv(*input)) { if (IsVariable(*input) || IsRecv(*input)) {
return false; return false;
} else if (IsSwitch(*input)) {
// Don't turn Identity nodes following Switch into NoOp or remove them
// if it requires anchoring a control dependencies the Switch node, which
// is not valid.
if (str_util::StartsWith(node.name(), kConstantFoldingCtrl)) {
// TODO(rmlarsen): Try to remove this artificial contraint.
return false;
} }
} for (const auto& consumer : node_map_->GetOutputs(node.name())) {
for (auto consumer : node_map_->GetOutputs(node.name())) {
if (node.input_size() > 1 && IsMerge(*consumer)) { if (node.input_size() > 1 && IsMerge(*consumer)) {
return false; return false;
} }