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/profiler | |
| parent | 91d98f5403145ad5899ecdaa8a6564da9bd111c9 (diff) | |
Reduce Grappler overhead by skipping optimizers when the graph is tiny.
PiperOrigin-RevId: 201095811
Diffstat (limited to 'tensorflow/python/profiler')
| -rw-r--r-- | tensorflow/python/profiler/model_analyzer_test.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/profiler/model_analyzer_test.py b/tensorflow/python/profiler/model_analyzer_test.py index 9e49188c1e..f9891f3b1e 100644 --- a/tensorflow/python/profiler/model_analyzer_test.py +++ b/tensorflow/python/profiler/model_analyzer_test.py @@ -707,8 +707,10 @@ class PrintModelAnalysisTest(test.TestCase): a = array_ops.constant(np.ones((100, 100))) b = array_ops.constant(np.ones((100, 100))) c = a * b + config = config_pb2.ConfigProto() + config.graph_options.rewrite_options.min_graph_nodes = -1 - with session.Session() as sess: + with session.Session(config=config) as sess: run_options = config_pb2.RunOptions( trace_level=config_pb2.RunOptions.FULL_TRACE) run_metadata = config_pb2.RunMetadata() |