diff options
| -rw-r--r-- | tensorflow/core/grappler/optimizers/pin_to_host_optimizer.cc | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.cc b/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.cc index 2190d38937..89eb76046e 100644 --- a/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.cc @@ -169,7 +169,13 @@ bool IsTPUGraphDef(const GraphDef& def) { } // All the nodes that should be blacklisted and not swapped. -bool IsBlacklisted(const NodeDef& node) { return IsCollective(node); } +bool IsBlacklisted(const NodeDef& node) { + return + // Collective ops should not be swapped. + IsCollective(node) || + // NoOp breaks perf regression tests (probably due to group dependencies). + IsNoOp(node); +} } // end namespace internal Status PinToHostOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, |