aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/tools/BUILD
diff options
context:
space:
mode:
authorGravatar Nupur Garg <nupurgarg@google.com>2018-04-11 09:34:51 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-11 09:44:18 -0700
commitadfbc272ded60a221444423b1fee58551c6445c7 (patch)
treed7125ca8d79b23c9802757fadd816ba781a6b391 /tensorflow/python/tools/BUILD
parent0073d1375add58b0493449c356af76aa33455f7d (diff)
Fixing dependencies.
PiperOrigin-RevId: 192461382
Diffstat (limited to 'tensorflow/python/tools/BUILD')
-rw-r--r--tensorflow/python/tools/BUILD14
1 files changed, 4 insertions, 10 deletions
diff --git a/tensorflow/python/tools/BUILD b/tensorflow/python/tools/BUILD
index 6e39ce8c80..cc2884a4f6 100644
--- a/tensorflow/python/tools/BUILD
+++ b/tensorflow/python/tools/BUILD
@@ -28,7 +28,7 @@ py_library(
name = "saved_model_utils",
srcs = ["saved_model_utils.py"],
srcs_version = "PY2AND3",
- deps = ["//tensorflow:tensorflow_py"],
+ deps = ["//tensorflow/contrib/saved_model:reader"],
)
py_library(
@@ -38,11 +38,12 @@ py_library(
deps = [
":saved_model_utils",
"//tensorflow/core:protos_all_py",
- "//tensorflow/python", # TODO(b/34059704): remove when fixed
"//tensorflow/python:client",
"//tensorflow/python:framework",
+ "//tensorflow/python:parsing_ops",
"//tensorflow/python:platform",
"//tensorflow/python:training",
+ "//tensorflow/python/saved_model:loader",
"@six_archive//:six",
],
)
@@ -52,14 +53,7 @@ py_binary(
srcs = ["freeze_graph.py"],
srcs_version = "PY2AND3",
deps = [
- ":saved_model_utils",
- "//tensorflow/core:protos_all_py",
- "//tensorflow/python", # TODO(b/34059704): remove when fixed
- "//tensorflow/python:client",
- "//tensorflow/python:framework",
- "//tensorflow/python:platform",
- "//tensorflow/python:training",
- "@six_archive//:six",
+ ":freeze_graph_lib",
],
)