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/core/kernels/data/experimental/BUILD | |
parent | 12e164d1e7c0b197f06d5d3c2ed26318b89b5e4c (diff) |
[tf.data] NUMA-aware MapAndBatch dataset.
PiperOrigin-RevId: 216395709
Diffstat (limited to 'tensorflow/core/kernels/data/experimental/BUILD')
-rw-r--r-- | tensorflow/core/kernels/data/experimental/BUILD | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/data/experimental/BUILD b/tensorflow/core/kernels/data/experimental/BUILD index 43406db3ed..4cf5643bc0 100644 --- a/tensorflow/core/kernels/data/experimental/BUILD +++ b/tensorflow/core/kernels/data/experimental/BUILD @@ -103,6 +103,22 @@ tf_kernel_library( ) tf_kernel_library( + name = "numa_map_and_batch_dataset_op", + srcs = ["numa_map_and_batch_dataset_op.cc"], + deps = [ + "//tensorflow/core:core_cpu_internal", + "//tensorflow/core:experimental_dataset_ops_op_lib", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core/kernels:inplace_ops", + "//tensorflow/core/kernels/data:captured_function", + "//tensorflow/core/kernels/data:dataset", + "@com_google_absl//absl/memory", + ], +) + +tf_kernel_library( name = "unique_dataset_op", srcs = ["unique_dataset_op.cc"], deps = [ @@ -132,6 +148,7 @@ tf_kernel_library( ":ignore_errors_dataset_op", ":indexed_dataset", ":lmdb_dataset_op", + ":numa_map_and_batch_dataset_op", ":prefetching_kernels", ":threadpool_dataset_op", ":unique_dataset_op", |