diff options
author | Yao Zhang <yaozhang@google.com> | 2018-08-27 15:11:13 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-27 15:14:54 -0700 |
commit | ab8e195d2e0978c21234a5632d4fabf47535eda1 (patch) | |
tree | 6145d69d652d9503b9a430d0b2a1a3f167e7139d /tensorflow/python/BUILD | |
parent | 599bb5522d622b2f4feaee06b5dd194b04424345 (diff) |
Open source graph analyzer.
PiperOrigin-RevId: 210439649
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index f62a927925..5af6437c56 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3777,6 +3777,7 @@ tf_py_wrap_cc( "framework/python_op_gen.i", "grappler/cluster.i", "grappler/cost_analyzer.i", + "grappler/graph_analyzer.i", "grappler/item.i", "grappler/model_analyzer.i", "grappler/tf_optimizer.i", @@ -3835,6 +3836,7 @@ tf_py_wrap_cc( "//tensorflow/core/grappler/clusters:single_machine", "//tensorflow/core/grappler/clusters:virtual_cluster", "//tensorflow/core/grappler/costs:graph_memory", + "//tensorflow/core/grappler/graph_analyzer:graph_analyzer_tool", "//tensorflow/core/grappler/optimizers:meta_optimizer", "//tensorflow/core:lib", "//tensorflow/core:reader_base", @@ -5536,6 +5538,18 @@ py_test( ], ) +py_binary( + name = "graph_analyzer", + srcs = [ + "grappler/graph_analyzer.py", + ], + srcs_version = "PY2AND3", + deps = [ + ":framework_for_generated_wrappers", + ":pywrap_tensorflow_internal", + ], +) + pyx_library( name = "framework_fast_tensor_util", srcs = ["framework/fast_tensor_util.pyx"], |