diff options
| -rw-r--r-- | tensorflow/core/grappler/optimizers/meta_optimizer.cc | 6 | ||||
| -rw-r--r-- | tensorflow/core/protobuf/rewriter_config.proto | 2 |
2 files changed, 8 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 || diff --git a/tensorflow/core/protobuf/rewriter_config.proto b/tensorflow/core/protobuf/rewriter_config.proto index 482178a540..8e0448d536 100644 --- a/tensorflow/core/protobuf/rewriter_config.proto +++ b/tensorflow/core/protobuf/rewriter_config.proto @@ -77,6 +77,8 @@ message RewriterConfig { Toggle scoped_allocator_optimization = 15; // Force small ops onto the CPU (default is ON). Toggle pin_to_host_optimization = 18; + // Disable the entire meta optimizer (off by default). + bool disable_meta_optimizer = 19; // Controls how many times we run the optimizers in meta optimizer (default // is once). |