aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/profiler
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-08-16 18:56:07 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-08-16 19:00:02 -0700
commite5e91b4a25fb40df09a7ff9f9a4ab148e38c7d1c (patch)
tree170be7c80f72901c3e6bb5db06997a9600d311da /tensorflow/python/profiler
parentf119dd4369f14017a3aa9ba00a7839df7976d5b1 (diff)
BUILD cleanup in tensorflow/python/...
PiperOrigin-RevId: 165530192
Diffstat (limited to 'tensorflow/python/profiler')
-rw-r--r--tensorflow/python/profiler/BUILD11
-rw-r--r--tensorflow/python/profiler/internal/BUILD7
2 files changed, 9 insertions, 9 deletions
diff --git a/tensorflow/python/profiler/BUILD b/tensorflow/python/profiler/BUILD
index f9e8578866..32ecde0243 100644
--- a/tensorflow/python/profiler/BUILD
+++ b/tensorflow/python/profiler/BUILD
@@ -13,7 +13,9 @@ py_library(
visibility = ["//visibility:public"],
deps = [
":model_analyzer",
+ ":option_builder",
":tfprof_logger",
+ "//tensorflow/core/profiler:protos_all_py",
"//tensorflow/python:util",
],
)
@@ -35,7 +37,9 @@ py_library(
":option_builder",
":tfprof_logger",
"//tensorflow/core/profiler:protos_all_py",
+ "//tensorflow/python:errors",
"//tensorflow/python:pywrap_tensorflow",
+ "@six_archive//:six",
],
)
@@ -124,10 +128,11 @@ py_test(
tags = ["no_pip"], # TODO(annarev): get it working with pip.
deps = [
":pprof_profiler",
- "//tensorflow/python:client",
+ "//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
- "//tensorflow/python:framework_test_lib",
- "//tensorflow/python:platform_test",
+ "//tensorflow/python:constant_op",
+ "//tensorflow/python:control_flow_ops",
+ "//tensorflow/python:math_ops",
"@com_google_pprof//:pprof_proto_py",
],
)
diff --git a/tensorflow/python/profiler/internal/BUILD b/tensorflow/python/profiler/internal/BUILD
index 5a6e53121c..f9cc8af19d 100644
--- a/tensorflow/python/profiler/internal/BUILD
+++ b/tensorflow/python/profiler/internal/BUILD
@@ -12,20 +12,17 @@ py_library(
srcs = ["model_analyzer_testlib.py"],
srcs_version = "PY2AND3",
deps = [
- "//tensorflow/core:protos_all_py",
"//tensorflow/python:array_ops",
"//tensorflow/python:framework_for_generated_wrappers",
"//tensorflow/python:init_ops",
"//tensorflow/python:math_ops",
"//tensorflow/python:nn_grad",
"//tensorflow/python:nn_ops",
- "//tensorflow/python:platform",
"//tensorflow/python:rnn",
+ "//tensorflow/python:rnn_cell",
"//tensorflow/python:tensor_array_grad",
"//tensorflow/python:training",
"//tensorflow/python:variable_scope",
- "//tensorflow/python:variables",
- "//tensorflow/python/profiler:model_analyzer",
],
)
@@ -39,10 +36,8 @@ py_test(
"//tensorflow/python:client",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_for_generated_wrappers",
- "//tensorflow/python:framework_test_lib",
"//tensorflow/python:init_ops",
"//tensorflow/python:nn_ops",
- "//tensorflow/python:platform_test",
"//tensorflow/python:pywrap_tensorflow",
"//tensorflow/python:variable_scope",
],