BUILD cleanup in tensorflow/tools/...

PiperOrigin-RevId: 160018623

tensorflow/tools/api/lib/BUILD

@@ -22,7 +22,8 @@ py_library(
     srcs_version = "PY2AND3",
     deps = [
         ":api_objects_proto_py",
-        "//tensorflow/tools/common:traverse",
+        "//tensorflow/python:platform",
+        "//tensorflow/python:util",
     ],
 )
 

tensorflow/tools/api/tests/BUILD

@@ -23,11 +23,12 @@ py_test(
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow:tensorflow_py",
+        "//tensorflow/python:client_testlib",
+        "//tensorflow/python:lib",
         "//tensorflow/python:platform",
         "//tensorflow/tools/api/lib:python_object_to_proto_visitor",
         "//tensorflow/tools/common:public_api",
         "//tensorflow/tools/common:traverse",
-        "@protobuf//:protobuf_python",
     ],
 )
 

tensorflow/tools/common/BUILD

@@ -15,6 +15,7 @@ py_library(
     name = "public_api",
     srcs = ["public_api.py"],
     srcs_version = "PY2AND3",
+    deps = ["//tensorflow/python:util"],
 )
 
 py_test(
@@ -32,6 +33,7 @@ py_library(
     name = "traverse",
     srcs = ["traverse.py"],
     srcs_version = "PY2AND3",
+    deps = ["//tensorflow/python:util"],
 )
 
 py_test(

tensorflow/tools/compatibility/BUILD

@@ -24,7 +24,9 @@ py_test(
     srcs_version = "PY2AND3",
     deps = [
         "tf_upgrade",
-        "//tensorflow:tensorflow_py",
+        "//tensorflow/python:client_testlib",
+        "//tensorflow/python:framework_test_lib",
+        "@six_archive//:six",
     ],
 )
 

tensorflow/tools/dist_test/scripts/BUILD

@@ -17,6 +17,6 @@ py_test(
     srcs_version = "PY2AND3",
     deps = [
         ":k8s_tensorflow_lib",
-        "//tensorflow/python:client_testlib",
+        "//tensorflow/python:platform_test",
     ],
 )

tensorflow/tools/graph_transforms/BUILD

@@ -240,7 +240,6 @@ cc_binary(
     deps = [
         ":transform_utils",
         "//tensorflow/core:core_cpu_internal",
-        "//tensorflow/core:framework",
         "//tensorflow/core:framework_internal",
         "//tensorflow/core:lib",
     ],
@@ -250,7 +249,12 @@ py_library(
     name = "transform_graph_py",
     srcs = ["__init__.py"],
     srcs_version = "PY2AND3",
-    deps = ["//tensorflow/python:pywrap_tensorflow"],
+    deps = [
+        "//tensorflow/core:protos_all_py",
+        "//tensorflow/python:errors",
+        "//tensorflow/python:pywrap_tensorflow",
+        "//tensorflow/python:util",
+    ],
 )
 
 tf_py_test(

tensorflow/tools/quantization/BUILD

@@ -13,7 +13,20 @@ py_library(
     name = "quantize_graph_lib",
     srcs = ["quantize_graph.py"],
     srcs_version = "PY2AND3",
-    deps = ["//tensorflow:tensorflow_py"],
+    deps = [
+        "//tensorflow/core:protos_all_py",
+        "//tensorflow/python:array_ops",
+        "//tensorflow/python:constant_op",
+        "//tensorflow/python:dtypes",
+        "//tensorflow/python:framework",
+        "//tensorflow/python:framework_ops",
+        "//tensorflow/python:graph_util",
+        "//tensorflow/python:platform",
+        "//tensorflow/python:session",
+        "//tensorflow/python:tensor_shape",
+        "//tensorflow/python:tensor_util",
+        "//third_party/py/numpy",
+    ],
 )
 
 py_binary(
@@ -27,18 +40,17 @@ py_binary(
         "//tensorflow/python:client",
         "//tensorflow/python:framework",
         "//tensorflow/python:framework_for_generated_wrappers",
+        "//tensorflow/python:graph_util",
         "//tensorflow/python:platform",
+        "//tensorflow/python:tensor_util",
         "//third_party/py/numpy",
-        "@six_archive//:six",
     ],
 )
 
 py_test(
     name = "quantize_graph_test",
     size = "small",
-    srcs = [
-        "quantize_graph_test.py",
-    ],
+    srcs = ["quantize_graph_test.py"],
     srcs_version = "PY2AND3",
     tags = ["nomsan"],  # http://b/32242946
     deps = [
@@ -48,6 +60,7 @@ py_test(
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:framework",
         "//tensorflow/python:framework_for_generated_wrappers",
+        "//tensorflow/python:graph_util",
         "//tensorflow/python:platform",
         "//third_party/py/numpy",
     ],
@@ -55,12 +68,13 @@ py_test(
 
 py_binary(
     name = "graph_to_dot",
-    srcs = [
-        "graph_to_dot.py",
-    ],
+    srcs = ["graph_to_dot.py"],
     main = "graph_to_dot.py",
     srcs_version = "PY2AND3",
-    deps = ["//tensorflow/python:platform"],
+    deps = [
+        "//tensorflow/core:protos_all_py",
+        "//tensorflow/python:platform",
+    ],
 )
 
 filegroup(

tensorflow/tools/test/BUILD

@@ -22,6 +22,7 @@ py_library(
     ],
     srcs_version = "PY2AND3",
     deps = [
+        "//tensorflow/core:protos_all_py",
         "//tensorflow/python:client",
         "//tensorflow/python:errors",
         "//tensorflow/python:platform",
@@ -46,6 +47,7 @@ py_library(
     srcs_version = "PY2AND3",
     deps = [
         ":system_info_lib",
+        "//tensorflow/core:protos_all_py",
         "//tensorflow/python:platform",
     ],
 )
@@ -57,6 +59,7 @@ py_binary(
     visibility = ["//visibility:public"],
     deps = [
         ":run_and_gather_logs_lib",
+        "//tensorflow/core:protos_all_py",
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:platform",
     ],

tensorflow/tools/tfprof/internal/BUILD

@@ -41,7 +41,6 @@ cc_library(
         ":tfprof_utils",
         "//tensorflow/core:lib",
         "//tensorflow/core:protos_all_cc",
-        "//tensorflow/tools/tfprof:protos_all_cc",
         "@jsoncpp_git//:jsoncpp",
     ],
 )
@@ -56,6 +55,7 @@ cc_library(
         "//tensorflow/core:lib",
         "//tensorflow/core:protos_all_cc",
         "//tensorflow/core:regexp_internal",
+        "//tensorflow/tools/tfprof:protos_all_cc",
     ],
 )
 
@@ -92,6 +92,7 @@ cc_library(
         ":tfprof_tensor",
         ":tfprof_utils",
         "//tensorflow/c:c_api",
+        "//tensorflow/c:checkpoint_reader",
         "//tensorflow/core:framework",
         "//tensorflow/core:lib",
         "//tensorflow/core:protos_all_cc",
@@ -264,9 +265,7 @@ cc_library(
 
 cc_library(
     name = "print_model_analysis_hdr",
-    hdrs = [
-        "print_model_analysis.h",
-    ],
+    hdrs = ["print_model_analysis.h"],
     deps = [
         "//tensorflow/core:framework_lite",
         "//tensorflow/core:protos_all_cc",

tensorflow/tools/tfprof/internal/advisor/BUILD

@@ -14,6 +14,7 @@ cc_library(
     name = "checker",
     hdrs = ["checker.h"],
     deps = [
+        "//tensorflow/core:lib",
         "//tensorflow/tools/tfprof/internal:tfprof_stats",
     ],
 )
@@ -23,7 +24,7 @@ cc_library(
     srcs = ["internal_checker_runner_dummy.cc"],
     hdrs = ["internal_checker_runner.h"],
     deps = [
-        ":checker",
+        "//tensorflow/tools/tfprof/internal:tfprof_utils",
     ],
 )