diff options
Diffstat (limited to 'tensorflow/core/grappler/utils/grappler_test.cc')
-rw-r--r-- | tensorflow/core/grappler/utils/grappler_test.cc | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/core/grappler/utils/grappler_test.cc b/tensorflow/core/grappler/utils/grappler_test.cc index 910b0acaef..6266733f3e 100644 --- a/tensorflow/core/grappler/utils/grappler_test.cc +++ b/tensorflow/core/grappler/utils/grappler_test.cc @@ -30,13 +30,16 @@ GrapplerTest::GrapplerTest() { // optimizations interfering in the comparison. RewriterConfig* cfg = options_.config.mutable_graph_options()->mutable_rewrite_options(); - cfg->set_constant_folding(RewriterConfig::OFF); + // TODO(rmlarsen): Add utility to generate config w/ all optimizers turned + // off. cfg->set_arithmetic_optimization(RewriterConfig::OFF); + cfg->set_constant_folding(RewriterConfig::OFF); + cfg->set_debug_stripper(RewriterConfig::OFF); cfg->set_dependency_optimization(RewriterConfig::OFF); - cfg->set_loop_optimization(RewriterConfig::OFF); cfg->set_function_optimization(RewriterConfig::OFF); cfg->set_layout_optimizer(RewriterConfig::OFF); - cfg->set_debug_stripper(RewriterConfig::OFF); + cfg->set_loop_optimization(RewriterConfig::OFF); + cfg->set_pin_to_host_optimization(RewriterConfig::OFF); } std::vector<Tensor> GrapplerTest::EvaluateNodes( |