diff options
Diffstat (limited to 'tensorflow/core/kernels/data/BUILD')
-rw-r--r-- | tensorflow/core/kernels/data/BUILD | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/data/BUILD b/tensorflow/core/kernels/data/BUILD index 3a1ac73f64..b3c359010d 100644 --- a/tensorflow/core/kernels/data/BUILD +++ b/tensorflow/core/kernels/data/BUILD @@ -675,6 +675,19 @@ tf_kernel_library( ) tf_kernel_library( + name = "model_dataset_op", + srcs = ["model_dataset_op.cc"], + deps = [ + ":dataset", + "//tensorflow/core:core_cpu_internal", + "//tensorflow/core:dataset_ops_op_lib", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + ], +) + +tf_kernel_library( name = "dataset_ops", srcs = ["dataset_ops.cc"], deps = [ @@ -708,6 +721,7 @@ tf_kernel_library( ":map_and_batch_dataset_op", ":map_dataset_op", ":map_defun_op", + ":model_dataset_op", ":optimize_dataset_op", ":optional_ops", ":padded_batch_dataset_op", |