diff options
Diffstat (limited to 'tensorflow/core/kernels/data/BUILD')
-rw-r--r-- | tensorflow/core/kernels/data/BUILD | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/data/BUILD b/tensorflow/core/kernels/data/BUILD index b3c359010d..6333853cdf 100644 --- a/tensorflow/core/kernels/data/BUILD +++ b/tensorflow/core/kernels/data/BUILD @@ -628,6 +628,20 @@ tf_kernel_library( ) tf_kernel_library( + name = "multi_device_iterator_ops", + srcs = ["multi_device_iterator_ops.cc"], + deps = [ + ":dataset", + ":dataset_utils", + "//tensorflow/core:core_cpu_internal", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core/kernels:ops_util", + ], +) + +tf_kernel_library( name = "optional_ops", srcs = ["optional_ops.cc"], hdrs = ["optional_ops.h"], @@ -722,6 +736,7 @@ tf_kernel_library( ":map_dataset_op", ":map_defun_op", ":model_dataset_op", + ":multi_device_iterator_ops", ":optimize_dataset_op", ":optional_ops", ":padded_batch_dataset_op", @@ -750,6 +765,7 @@ tf_kernel_library( ":window_dataset_op", ":writer_ops", ":zip_dataset_op", + "//tensorflow/core/kernels/data/experimental:dataset_kernels", ], ) |