diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2018-04-26 13:19:39 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-26 13:21:59 -0700 |
| commit | 2c105ace934edce193669b55b13b64283caa24d7 (patch) | |
| tree | 068bbe4e4b2a43557b49d896e524cff8d5090a16 /tensorflow/python/grappler | |
| parent | d3c18b5dcf5293e81bfd9acdce3a3b8f79ae4ade (diff) | |
Run 2 passes of rewrites by default
PiperOrigin-RevId: 194443770
Diffstat (limited to 'tensorflow/python/grappler')
| -rw-r--r-- | tensorflow/python/grappler/memory_optimizer_test.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/grappler/memory_optimizer_test.py b/tensorflow/python/grappler/memory_optimizer_test.py index 4df959ce04..3f9d8864a2 100644 --- a/tensorflow/python/grappler/memory_optimizer_test.py +++ b/tensorflow/python/grappler/memory_optimizer_test.py @@ -76,6 +76,7 @@ class MemoryOptimizerSwapTest(test.TestCase): rewriter_config = rewriter_config_pb2.RewriterConfig( disable_model_pruning=True, + meta_optimizer_iterations=rewriter_config_pb2.RewriterConfig.ONE, constant_folding=rewriter_config_pb2.RewriterConfig.OFF, memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL) graph = tf_optimizer.OptimizeGraph(rewriter_config, mg) |