diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-06 03:31:42 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-06 03:36:01 -0700 |
| commit | 3ef0e4eba6013d96f761bece3c2f724ad701c2d2 (patch) | |
| tree | 05be0bf4dd9c3e157f96fcc6c8eafa96e957a7cf /tensorflow/contrib/batching | |
| parent | f9d3e8e29e1e8bcb2cc254b5af2c523776a41713 (diff) | |
Further BUILD cleanup in tensorflow/contrib/...
PiperOrigin-RevId: 161061704
Diffstat (limited to 'tensorflow/contrib/batching')
| -rw-r--r-- | tensorflow/contrib/batching/BUILD | 9 | ||||
| -rw-r--r-- | tensorflow/contrib/batching/util/BUILD | 1 |
2 files changed, 6 insertions, 4 deletions
diff --git a/tensorflow/contrib/batching/BUILD b/tensorflow/contrib/batching/BUILD index acace9b3ba..3cf8c318eb 100644 --- a/tensorflow/contrib/batching/BUILD +++ b/tensorflow/contrib/batching/BUILD @@ -32,6 +32,7 @@ cc_test( deps = [ ":batch_scheduler", "//tensorflow/core:lib", + "//tensorflow/core:test", "//tensorflow/core:test_main", ], ) @@ -52,6 +53,7 @@ cc_library( deps = [ ":batch_scheduler", ":shared_batch_scheduler_hdrs", + "//tensorflow/contrib/batching/util:periodic_function_dynamic", "//tensorflow/core:lib", ], alwayslink = 1, @@ -187,12 +189,11 @@ py_test( ], deps = [ ":batch_py", - "//tensorflow/python:framework_test_lib", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:dtypes", "//tensorflow/python:gradients", - "//tensorflow/python:platform_test", "//tensorflow/python:script_ops", - "//tensorflow/python:state_ops", - "//tensorflow/python:variables", ], ) diff --git a/tensorflow/contrib/batching/util/BUILD b/tensorflow/contrib/batching/util/BUILD index 55dd898aa2..21dd26722a 100644 --- a/tensorflow/contrib/batching/util/BUILD +++ b/tensorflow/contrib/batching/util/BUILD @@ -46,6 +46,7 @@ cc_test( deps = [ ":periodic_function", "//tensorflow/contrib/batching/test_util:fake_clock_env", + "//tensorflow/core:test", "//tensorflow/core:test_main", ], ) |