diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-18 18:22:04 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-18 18:25:01 -0700 |
| commit | 27acbe0b4c7f13d52762419d2d819b11c1d9f54b (patch) | |
| tree | 2378a5e07b206cd69feab40425b1e70d0ee8b047 /tensorflow/python/debug | |
| parent | 91d98f5403145ad5899ecdaa8a6564da9bd111c9 (diff) | |
Reduce Grappler overhead by skipping optimizers when the graph is tiny.
PiperOrigin-RevId: 201095811
Diffstat (limited to 'tensorflow/python/debug')
| -rw-r--r-- | tensorflow/python/debug/lib/debug_graph_reconstruction_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/debug/lib/debug_graph_reconstruction_test.py b/tensorflow/python/debug/lib/debug_graph_reconstruction_test.py index bd00f73861..676097fde9 100644 --- a/tensorflow/python/debug/lib/debug_graph_reconstruction_test.py +++ b/tensorflow/python/debug/lib/debug_graph_reconstruction_test.py @@ -44,7 +44,8 @@ class ReconstructNonDebugGraphTest(test_util.TensorFlowTestCase): def _no_rewrite_session_config(self): rewriter_config = rewriter_config_pb2.RewriterConfig( - dependency_optimization=rewriter_config_pb2.RewriterConfig.OFF) + dependency_optimization=rewriter_config_pb2.RewriterConfig.OFF, + min_graph_nodes=-1) graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config) return config_pb2.ConfigProto(graph_options=graph_options) |