Add a few missing inclusions

PiperOrigin-RevId: 165049580

3 files changed, 14 insertions, 7 deletions

diff --git a/tensorflow/core/kernels/reference_gemm.h b/tensorflow/core/kernels/reference_gemm.h
index ad8f427429..bb2a21720f 100644
--- a/tensorflow/core/kernels/reference_gemm.h
+++ b/tensorflow/core/kernels/reference_gemm.h
@@ -16,6 +16,11 @@ limitations under the License.
 #ifndef THIRD_PARTY_TENSORFLOW_CORE_KERNELS_REFERENCE_GEMM_H_
 #define THIRD_PARTY_TENSORFLOW_CORE_KERNELS_REFERENCE_GEMM_H_
 
+#include <stdlib.h>
+
+#include "third_party/eigen3/Eigen/Core"
+#include "tensorflow/core/platform/types.h"
+
 // This is an unoptimized but debuggable implementation of the GEMM matrix
 // multiply function, used to compare to faster but more opaque versions, or
 // for bit depths or argument combinations that aren't supported by optimized
diff --git a/tensorflow/core/lib/io/zlib_compression_options.h b/tensorflow/core/lib/io/zlib_compression_options.h
index b1e5875843..dc7218e866 100644
--- a/tensorflow/core/lib/io/zlib_compression_options.h
+++ b/tensorflow/core/lib/io/zlib_compression_options.h
@@ -18,8 +18,11 @@ limitations under the License.
 
 #include <zlib.h>
 
+#include "tensorflow/core/platform/types.h"
+
 namespace tensorflow {
 namespace io {
+
 class ZlibCompressionOptions {
  public:
   static ZlibCompressionOptions DEFAULT();
@@ -121,7 +124,8 @@ inline ZlibCompressionOptions ZlibCompressionOptions::GZIP() {
   options.window_bits = options.window_bits + 16;
   return options;
 }
-}
-}
+
+}  // namespace io
+}  // namespace tensorflow
 
 #endif  // TENSORFLOW_LIB_IO_ZLIB_COMPRESSION_OPTIONS_H_
diff --git a/tensorflow/core/util/tensor_bundle/BUILD b/tensorflow/core/util/tensor_bundle/BUILD
index 1045cf84b4..4421925c9e 100644
--- a/tensorflow/core/util/tensor_bundle/BUILD
+++ b/tensorflow/core/util/tensor_bundle/BUILD
@@ -1,9 +1,8 @@
 # Description:
-# Tensor bundle: a module to efficiently serialize and deserialize tensors.
+#   Tensor bundle: a module to efficiently serialize and deserialize tensors.
 
 package(
     default_visibility = ["//visibility:public"],
-    features = ["-parse_headers"],
 )
 
 licenses(["notice"])  # Apache 2.0
@@ -50,9 +49,8 @@ cc_library(
 
 cc_header_only_library(
     name = "tensor_bundle_headers_lib",
-    deps = [
-        ":tensor_bundle",
-    ],
+    features = ["-parse_headers"],  # Transitively pulls in Eigen headers
+    deps = [":tensor_bundle"],
 )
 
 cc_library(