aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/BUILD
diff options
context:
space:
mode:
authorGravatar Yao Zhang <yaozhang@google.com>2018-08-27 15:11:13 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-27 15:14:54 -0700
commitab8e195d2e0978c21234a5632d4fabf47535eda1 (patch)
tree6145d69d652d9503b9a430d0b2a1a3f167e7139d /tensorflow/python/BUILD
parent599bb5522d622b2f4feaee06b5dd194b04424345 (diff)
Open source graph analyzer.
PiperOrigin-RevId: 210439649
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r--tensorflow/python/BUILD14
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"],