diff options
author | Rachel Lim <rachelim@google.com> | 2018-08-21 15:52:10 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 15:56:26 -0700 |
commit | 62fcb03449c13281935a154e3ea7c11614ffb678 (patch) | |
tree | 4f60de30aaf9cc243d57d139ff11296248800d60 /tensorflow/tools/pip_package | |
parent | 7989b2bc99aa5207dccae9e7bedfc3cb140349cd (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/BUILD | 1 |
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", |