diff options
-rw-r--r-- | tensorflow/core/kernels/reference_gemm.h | 5 | ||||
-rw-r--r-- | tensorflow/core/lib/io/zlib_compression_options.h | 8 | ||||
-rw-r--r-- | tensorflow/core/util/tensor_bundle/BUILD | 8 |
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( |