diff options
author | Brennan Saeta <saeta@google.com> | 2018-10-09 11:54:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 11:58:43 -0700 |
commit | 072fcb995a3fd658ee2461b59b159498c710513d (patch) | |
tree | f3def3d3ac6e270ad32e428889a79d662c8bc9cf /tensorflow/python/data/experimental/kernel_tests/serialization/BUILD | |
parent | 12e164d1e7c0b197f06d5d3c2ed26318b89b5e4c (diff) |
[tf.data] NUMA-aware MapAndBatch dataset.
PiperOrigin-RevId: 216395709
Diffstat (limited to 'tensorflow/python/data/experimental/kernel_tests/serialization/BUILD')
-rw-r--r-- | tensorflow/python/data/experimental/kernel_tests/serialization/BUILD | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD index e556b65b7c..a97cff9fbb 100644 --- a/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD @@ -307,6 +307,21 @@ py_test( ) py_test( + name = "numa_map_and_batch_dataset_serialization_test", + size = "medium", + srcs = ["numa_map_and_batch_dataset_serialization_test.py"], + srcs_version = "PY2AND3", + tags = ["no_pip"], + deps = [ + ":dataset_serialization_test_base", + "//tensorflow/python:client_testlib", + "//tensorflow/python:math_ops", + "//tensorflow/python/data/experimental/ops:batching", + "//tensorflow/python/data/ops:dataset_ops", + ], +) + +py_test( name = "map_dataset_serialization_test", size = "medium", srcs = ["map_dataset_serialization_test.py"], |