diff options
author | 2018-04-11 09:34:51 -0700 | |
---|---|---|
committer | 2018-04-11 09:44:18 -0700 | |
commit | adfbc272ded60a221444423b1fee58551c6445c7 (patch) | |
tree | d7125ca8d79b23c9802757fadd816ba781a6b391 /tensorflow/python/tools/BUILD | |
parent | 0073d1375add58b0493449c356af76aa33455f7d (diff) |
Fixing dependencies.
PiperOrigin-RevId: 192461382
Diffstat (limited to 'tensorflow/python/tools/BUILD')
-rw-r--r-- | tensorflow/python/tools/BUILD | 14 |
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", ], ) |