Merge pull request #26632 from feihugis:Test_TakeDataset

PiperOrigin-RevId: 238291593
tensorflow · Mar 13, 2019 · de2bea7 · de2bea7
2 parents 790869c + 66b8ba4
commit de2bea7
Show file tree

Hide file tree

Showing 2 changed files with 580 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/data/BUILD b/tensorflow/core/kernels/data/BUILD
@@ -435,6 +435,26 @@ tf_kernel_library(
     ],
 )
 
+tf_cc_test(
+    name = "take_dataset_op_test",
+    size = "small",
+    srcs = ["take_dataset_op_test.cc"],
+    deps = [
+        ":dataset_test_base",
+        ":dataset_utils",
+        ":iterator_ops",
+        ":take_dataset_op",
+        ":tensor_slice_dataset_op",
+        "//tensorflow/core:core_cpu_internal",
+        "//tensorflow/core:dataset_ops_op_lib",
+        "//tensorflow/core:framework",
+        "//tensorflow/core:lib_internal",
+        "//tensorflow/core:test",
+        "//tensorflow/core:test_main",
+        "//tensorflow/core:testlib",
+    ],
+)
+
 tf_kernel_library(
     name = "skip_dataset_op",
     srcs = ["skip_dataset_op.cc"],