diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-28 16:31:05 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 16:35:07 -0700 |
| commit | 4eef4925853a284fdfd4b5fae4b65f594a883b3b (patch) | |
| tree | 27a2dbdfb33213552662046d86df0b50cfa0d207 /tensorflow/core/grappler | |
| parent | 478d370eb116ad2294134d75a886637a7d6da225 (diff) | |
Add a rewrite_config option to disable meta_optimizer.
PiperOrigin-RevId: 215014737
Diffstat (limited to 'tensorflow/core/grappler')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/meta_optimizer.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc index e18a5f21d2..406c1b60ce 100644 --- a/tensorflow/core/grappler/optimizers/meta_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/meta_optimizer.cc @@ -115,6 +115,9 @@ std::unique_ptr<GraphOptimizer> MetaOptimizer::MakeNewOptimizer( Status MetaOptimizer::InitializeOptimizers( std::vector<std::unique_ptr<GraphOptimizer>>* optimizers) const { + if (cfg_.disable_meta_optimizer()) { + return Status::OK(); + } if (!cfg_.disable_model_pruning()) { optimizers->push_back(MakeUnique<ModelPruner>()); } @@ -489,6 +492,9 @@ void MetaOptimizer::Feedback(Cluster* cluster, const GrapplerItem& item, } bool MetaOptimizerEnabled(const RewriterConfig& cfg) { + if (cfg.disable_meta_optimizer()) { + return false; + } return !cfg.disable_model_pruning() || cfg.layout_optimizer() != RewriterConfig::OFF || cfg.function_optimization() != RewriterConfig::OFF || |