Move tsl/protobuf/* besides error_codes.proto to xla/tsl/protobuf

`error_codes.proto` will be moved in a separate change

PiperOrigin-RevId: 678848832

tsl/platform/BUILD

@@ -323,7 +323,7 @@ cc_library(
     deps = [
         ":status",
         "//tsl/protobuf:error_codes_proto_impl_cc",
-        "//tsl/protobuf:status_proto_cc",
+        "@xla//xla/tsl/protobuf:status_proto_cc",
     ] + tf_platform_deps("status"),
 )
 
@@ -1349,10 +1349,10 @@ tsl_cc_test(
         ":test",
         ":test_main",
         "//tsl/protobuf:error_codes_proto_impl_cc",
-        "//tsl/protobuf:status_proto_cc",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings:cord",
         "@com_google_absl//absl/strings:str_format",
+        "@xla//xla/tsl/protobuf:status_proto_cc",
     ],
 )
 

tsl/platform/default/build_config.bzl

@@ -726,7 +726,7 @@ def tf_lib_proto_parsing_deps():
     return [
         ":protos_all_cc",
         clean_dep("@eigen_archive//:eigen3"),
-        clean_dep("//tsl/protobuf:protos_all_cc"),
+        clean_dep("@xla//xla/tsl/protobuf:protos_all_cc"),
     ]
 
 def tf_py_clif_cc(name, visibility = None, **kwargs):
@@ -779,8 +779,8 @@ def tsl_cc_test(
                 # TODO(ddunleavy) remove these and add proto deps to tests
                 # granularly
                 clean_dep("//tsl/protobuf:error_codes_proto_impl_cc_impl"),
-                clean_dep("//tsl/protobuf:histogram_proto_cc_impl"),
-                clean_dep("//tsl/protobuf:status_proto_cc_impl"),
+                clean_dep("@xla//xla/tsl/protobuf:histogram_proto_cc_impl"),
+                clean_dep("@xla//xla/tsl/protobuf:status_proto_cc_impl"),
                 clean_dep("//tsl/profiler/protobuf:xplane_proto_cc_impl"),
                 clean_dep("//tsl/profiler/protobuf:profiler_options_proto_cc_impl"),
             ],
@@ -789,7 +789,7 @@ def tsl_cc_test(
     )
 
 def tf_portable_proto_lib():
-    return ["//tensorflow/core:protos_all_cc_impl", clean_dep("//tsl/protobuf:protos_all_cc_impl")]
+    return ["//tensorflow/core:protos_all_cc_impl", clean_dep("@xla//xla/tsl/protobuf:protos_all_cc_impl")]
 
 def tf_protobuf_compiler_deps():
     return if_static(

tsl/platform/status_test.cc

@@ -18,13 +18,13 @@ limitations under the License.
 #include "absl/status/status.h"
 #include "absl/strings/cord.h"
 #include "absl/strings/str_format.h"
+#include "xla/tsl/protobuf/status.pb.h"
 #include "tsl/platform/errors.h"
 #include "tsl/platform/stack_frame.h"
 #include "tsl/platform/status_matchers.h"
 #include "tsl/platform/status_to_from_proto.h"
 #include "tsl/platform/test.h"
 #include "tsl/protobuf/error_codes.pb.h"
-#include "tsl/protobuf/status.pb.h"
 
 namespace tsl {
 namespace {

tsl/platform/status_to_from_proto.cc

@@ -16,9 +16,9 @@ limitations under the License.
 
 #include <string>
 
+#include "xla/tsl/protobuf/status.pb.h"
 #include "tsl/platform/status.h"
 #include "tsl/protobuf/error_codes.pb.h"
-#include "tsl/protobuf/status.pb.h"
 
 namespace tsl {
 

tsl/platform/status_to_from_proto.h

@@ -15,8 +15,8 @@ limitations under the License.
 #ifndef TENSORFLOW_TSL_PLATFORM_STATUS_TO_FROM_PROTO_H_
 #define TENSORFLOW_TSL_PLATFORM_STATUS_TO_FROM_PROTO_H_
 
+#include "xla/tsl/protobuf/status.pb.h"
 #include "tsl/platform/status.h"
-#include "tsl/protobuf/status.pb.h"
 
 namespace tsl {
 

tsl/protobuf/BUILD

@@ -1,4 +1,3 @@
-# Placeholder: load py_proto_library
 load(
     "@xla//xla/tsl:tsl.bzl",
     "if_google",
@@ -20,93 +19,10 @@ package(
     licenses = ["notice"],
 )
 
-tf_proto_library(
-    name = "dnn_proto",
-    srcs = ["dnn.proto"],
-    make_default_target_header_only = True,
-    protodeps = if_google(["//google/protobuf:wrappers"]),
-    visibility = ["//visibility:public"],
-)
-
 tf_proto_library(
     name = "error_codes_proto_impl",
     srcs = ["error_codes.proto"],
     make_default_target_header_only = True,
     protodeps = if_google(["//google/protobuf:any"]),
     visibility = ["//visibility:public"],
 )
-
-tf_proto_library(
-    name = "status_proto",
-    srcs = ["status.proto"],
-    make_default_target_header_only = True,
-    protodeps = [":error_codes_proto_impl"],
-    visibility = ["//visibility:public"],
-)
-
-tf_proto_library(
-    name = "histogram_proto",
-    srcs = ["histogram.proto"],
-    make_default_target_header_only = True,
-    visibility = ["//visibility:public"],
-)
-
-tf_proto_library(
-    name = "coordination_config_proto",
-    srcs = ["coordination_config.proto"],
-    make_default_target_header_only = True,
-    visibility = ["//visibility:public"],
-)
-
-tf_proto_library(
-    name = "coordination_service_proto",
-    srcs = ["coordination_service.proto"],
-    has_services = 1,
-    create_grpc_library = True,
-    create_java_proto = False,
-    create_service = True,
-    protodeps = if_google(["//google/protobuf:any"]),
-    visibility = ["//visibility:public"],
-)
-
-# copybara:uncomment_begin(google-only)
-# py_proto_library(
-#     name = "coordination_service_py_pb2",
-#     api_version = 2,
-#     visibility = ["//visibility:public"],
-#     deps = [":coordination_service_proto"],
-# )
-# copybara:uncomment_end
-
-tf_proto_library(
-    name = "distributed_runtime_payloads_proto",
-    srcs = ["distributed_runtime_payloads.proto"],
-    make_default_target_header_only = True,
-    visibility = ["//visibility:public"],
-)
-
-tf_proto_library(
-    name = "rpc_options_proto",
-    srcs = ["rpc_options.proto"],
-    make_default_target_header_only = True,
-    visibility = ["//visibility:public"],
-)
-
-tf_proto_library(
-    name = "protos_all",
-    create_go_proto = False,
-    make_default_target_header_only = True,
-    protodeps = [
-        # TODO(tlongeri): Conceptually, these fit into protos_all but adding them currently causes
-        # breakages (and they are not actually used).
-        "@xla//xla/tsl/protobuf:bfc_memory_map_proto",
-        ":coordination_config_proto",
-        ":distributed_runtime_payloads_proto",
-        ":error_codes_proto_impl",
-        ":histogram_proto",
-        ":rpc_options_proto",
-        ":status_proto",
-        "@xla//xla/tsl/protobuf:test_log_proto",
-    ] + if_google(["//google/protobuf:any"]),
-    visibility = ["//visibility:public"],
-)