aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/pip_package
diff options
context:
space:
mode:
authorGravatar Rachel Lim <rachelim@google.com>2018-08-21 15:52:10 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-21 15:56:26 -0700
commit62fcb03449c13281935a154e3ea7c11614ffb678 (patch)
tree4f60de30aaf9cc243d57d139ff11296248800d60 /tensorflow/tools/pip_package
parent7989b2bc99aa5207dccae9e7bedfc3cb140349cd (diff)
[tf.data] Add an optimization that vectorizes map functions and swaps the order of Map->Batch dataset transformations to Batch->Map
PiperOrigin-RevId: 209674669
Diffstat (limited to 'tensorflow/tools/pip_package')
-rw-r--r--tensorflow/tools/pip_package/BUILD1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/tools/pip_package/BUILD b/tensorflow/tools/pip_package/BUILD
index d98e5feae5..ac43f8449a 100644
--- a/tensorflow/tools/pip_package/BUILD
+++ b/tensorflow/tools/pip_package/BUILD
@@ -71,6 +71,7 @@ COMMON_PIP_DEPS = [
"//tensorflow/contrib/constrained_optimization:constrained_optimization_pip",
"//tensorflow/contrib/data/python/kernel_tests/serialization:dataset_serialization_test_base",
"//tensorflow/contrib/data/python/kernel_tests:stats_dataset_test_base",
+ "//tensorflow/contrib/data/python/kernel_tests:test_utils",
"//tensorflow/contrib/data/python/ops:contrib_op_loader",
"//tensorflow/contrib/eager/python/examples:examples_pip",
"//tensorflow/contrib/eager/python:evaluator",