diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-01 11:58:17 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 12:04:15 -0700 |
| commit | 694367b574dcaf5ac90f3e42b8dee8fa51ca9f38 (patch) | |
| tree | 117943578658c12ddf7e7af8bea89c21757f00eb /tensorflow/core/grappler | |
| parent | f1fd53748b99532b2572b8909efcd4f5c06ce28d (diff) | |
Automated rollback of commit cb98ceba9cff8c10ee3c7e89dc8925c88b28118e
PiperOrigin-RevId: 215254762
Diffstat (limited to 'tensorflow/core/grappler')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/meta_optimizer.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc index a5f851fb1a..c3d70a1fdf 100644 --- a/tensorflow/core/grappler/optimizers/meta_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/meta_optimizer.cc @@ -139,7 +139,7 @@ Status MetaOptimizer::InitializeOptimizers( if (cfg_.remapping() != RewriterConfig::OFF) { optimizers->push_back(MakeUnique<Remapper>(cfg_.remapping())); } - if (cfg_.pin_to_host_optimization() != RewriterConfig::OFF) { + if (cfg_.pin_to_host_optimization() == RewriterConfig::ON) { optimizers->push_back(MakeUnique<PinToHostOptimizer>()); } if (cfg_.arithmetic_optimization() != RewriterConfig::OFF) { @@ -527,7 +527,7 @@ bool MetaOptimizerEnabled(const RewriterConfig& cfg) { cfg.memory_optimization() != RewriterConfig::NO_MEM_OPT || cfg.debug_stripper() == RewriterConfig::ON || cfg.scoped_allocator_optimization() == RewriterConfig::ON || - cfg.pin_to_host_optimization() != RewriterConfig::OFF || + cfg.pin_to_host_optimization() == RewriterConfig::ON || !cfg.optimizers().empty() || !cfg.custom_optimizers().empty(); } |