diff options
author | Yao Zhang <yaozhang@google.com> | 2018-02-08 15:06:57 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-08 15:18:22 -0800 |
commit | a5d93698ad5ae1e6488b536abb8501cd6ec70551 (patch) | |
tree | 7138f3c678101dadb45acd239ea5757874f50f9a /tensorflow/python/grappler | |
parent | a00ba3780050b8e09f80788c56e3039da328c071 (diff) |
Add memory usage report to cost analyzer tool; run all default optimizations.
PiperOrigin-RevId: 185058999
Diffstat (limited to 'tensorflow/python/grappler')
-rw-r--r-- | tensorflow/python/grappler/cost_analyzer_tool.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/python/grappler/cost_analyzer_tool.py b/tensorflow/python/grappler/cost_analyzer_tool.py index ac251f2bbd..51b77b471b 100644 --- a/tensorflow/python/grappler/cost_analyzer_tool.py +++ b/tensorflow/python/grappler/cost_analyzer_tool.py @@ -54,14 +54,16 @@ def main(_): metagraph = saver.export_meta_graph( graph_def=graph.as_graph_def(), graph=graph) + rewriter_config = rewriter_config_pb2.RewriterConfig() if FLAGS.rewriter_config is not None: - rewriter_config = rewriter_config_pb2.RewriterConfig() text_format.Merge(FLAGS.rewriter_config, rewriter_config) - optimized_graph = tf_optimizer.OptimizeGraph(rewriter_config, metagraph) - metagraph.graph_def.CopyFrom(optimized_graph) + optimized_graph = tf_optimizer.OptimizeGraph(rewriter_config, metagraph) + metagraph.graph_def.CopyFrom(optimized_graph) report = cost_analyzer.GenerateCostReport(metagraph, FLAGS.per_node_report) print(report) + report = cost_analyzer.GenerateMemoryReport(metagraph) + print(report) if __name__ == "__main__": |