Add a number of missing headers being transitively pulled in.

This enables a few headers to be removed from implementations and in turn
simplify the build graph some.

PiperOrigin-RevId: 291428320
Change-Id: I1ac1b325eacf9909fcdf94230f328055177379d3

tensorflow/core/kernels/data/BUILD

@@ -121,6 +121,7 @@ cc_library(
         "//tensorflow/core/grappler/optimizers/data",
         "//tensorflow/core/grappler/optimizers/data:function_utils",
         "//tensorflow/core/grappler/optimizers/data:graph_utils",
+        "//tensorflow/core/kernels/data:serialization_utils",
     ],
 )
 
@@ -906,10 +907,9 @@ tf_cc_test(
     srcs = ["tensor_dataset_op_test.cc"],
     deps = [
         ":dataset_test_base",
+        ":dataset_utils",
         ":tensor_dataset_op",
-        "//tensorflow/core:test",
         "//tensorflow/core:test_main",
-        "//tensorflow/core:testlib",
     ],
 )
 

tensorflow/core/kernels/data/cache_dataset_ops_test.cc

@@ -12,6 +12,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/cache_dataset_ops.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/experimental/BUILD

@@ -33,13 +33,10 @@ tf_cc_test(
     deps = [
         ":assert_next_dataset_op",
         "//tensorflow/core:experimental_dataset_ops_op_lib",
-        "//tensorflow/core:framework",
-        "//tensorflow/core:test",
         "//tensorflow/core:test_main",
-        "//tensorflow/core:testlib",
         "//tensorflow/core/kernels/data:dataset_test_base",
+        "//tensorflow/core/kernels/data:range_dataset_op",
         "//tensorflow/core/kernels/data:take_dataset_op",
-        "//third_party/eigen3",
     ],
 )
 

tensorflow/core/kernels/data/experimental/assert_next_dataset_op_test.cc

@@ -12,6 +12,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/experimental/assert_next_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/range_dataset_op.h"
 #include "tensorflow/core/kernels/data/take_dataset_op.h"
 
 namespace tensorflow {

tensorflow/core/kernels/data/repeat_dataset_op_test.cc

@@ -15,6 +15,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/repeat_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/rewrite_utils.cc

@@ -31,6 +31,7 @@ limitations under the License.
 #include "tensorflow/core/grappler/optimizers/data/graph_utils.h"
 #include "tensorflow/core/grappler/optimizers/meta_optimizer.h"
 #include "tensorflow/core/kernels/data/dataset_utils.h"
+#include "tensorflow/core/kernels/data/serialization_utils.h"
 #include "tensorflow/core/lib/core/errors.h"
 #include "tensorflow/core/lib/hash/hash.h"
 #include "tensorflow/core/lib/strings/proto_serialization.h"

tensorflow/core/kernels/data/shuffle_dataset_op_test.cc

@@ -12,6 +12,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/shuffle_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/sparse_tensor_slice_dataset_op_test.cc

@@ -14,6 +14,7 @@ limitations under the License.
 ==============================================================================*/
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/tensor_dataset_op_test.cc

@@ -15,6 +15,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/tensor_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/tensor_slice_dataset_op_test.cc

@@ -15,6 +15,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/tensor_slice_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {

tensorflow/core/kernels/data/window_dataset_op_test.cc

@@ -12,6 +12,7 @@ limitations under the License.
 #include "tensorflow/core/kernels/data/window_dataset_op.h"
 
 #include "tensorflow/core/kernels/data/dataset_test_base.h"
+#include "tensorflow/core/kernels/data/dataset_utils.h"
 
 namespace tensorflow {
 namespace data {