diff --git a/tsl/platform/BUILD b/tsl/platform/BUILD index 1200daa03..a659254d5 100644 --- a/tsl/platform/BUILD +++ b/tsl/platform/BUILD @@ -198,6 +198,7 @@ cc_library( deps = [ ":platform", ":tstring", + "@com_google_absl//absl/base:core_headers", "@xla//xla/tsl/platform:types", ] + tsl_protobuf_deps(), ) diff --git a/tsl/platform/protobuf.h b/tsl/platform/protobuf.h index c2f8bee91..cd820320e 100644 --- a/tsl/platform/protobuf.h +++ b/tsl/platform/protobuf.h @@ -21,6 +21,7 @@ limitations under the License. #include #include +#include "absl/base/macros.h" #include "xla/tsl/platform/types.h" #include "tsl/platform/platform.h" @@ -57,8 +58,8 @@ limitations under the License. namespace tsl { namespace protobuf = ::google::protobuf; -using protobuf_int64 = int64_t; -using protobuf_uint64 = uint64_t; +using protobuf_int64 ABSL_DEPRECATE_AND_INLINE() = int64_t; +using protobuf_uint64 ABSL_DEPRECATE_AND_INLINE() = uint64_t; extern const char* kProtobufInt64Typename; extern const char* kProtobufUint64Typename;