diff --git a/tensorflow/core/protobuf/tpu/BUILD b/tensorflow/core/protobuf/tpu/BUILD
index 3bbb97885a0..a379b79a229 100644
--- a/tensorflow/core/protobuf/tpu/BUILD
+++ b/tensorflow/core/protobuf/tpu/BUILD
@@ -1,8 +1,8 @@
-load("//tensorflow/compiler/xla:xla.bzl", "xla_proto_library")
 load(
     "//tensorflow/core/platform:default/build_config.bzl",
     "tf_additional_all_protos",
     "tf_proto_library",
+    "tf_proto_library_cc",
     "tf_proto_library_py",
     "tf_pyclif_proto_library",
 )
@@ -74,18 +74,19 @@ tf_proto_library_py(
     visibility = ["//visibility:public"],
 )
 
-xla_proto_library(
+tf_proto_library_cc(
     name = "compile_metadata_proto",
     srcs = [
         "compile_metadata.proto",
     ],
-    visibility = ["//visibility:public"],
-    deps = [
+    cc_api_version = 2,
+    protodeps = [
         ":dynamic_padding_proto",
         "//tensorflow/compiler/xla:xla_data_proto",
         "//tensorflow/compiler/xla:xla_proto",
         "//tensorflow/core:protos_all",
     ],
+    visibility = ["//visibility:public"],
 )
 
 tf_pyclif_proto_library(