From c9ca63d25504e934fd808e3fc4898b4c5fbefda8 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 22 Feb 2024 03:02:56 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 609306481
---
 .../c/tf_rendezvous_c_api_helper.cc                 | 13 +++++++------
 .../c/tf_rendezvous_c_api_internal.cc               |  2 +-
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_helper.cc b/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_helper.cc
index f1d715ccf5c079..0750d3b6205946 100644
--- a/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_helper.cc
+++ b/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_helper.cc
@@ -101,9 +101,9 @@ using DoneCallbackParamPtr =
     std::unique_ptr<TF_RendezvousDoneCallback_Params, DoneCallbackParamDeleter>;
 
 SendParamDeleter MakeSendParamDeleter();
-StatusOr<SendParamPtr> SendParamsToC(const RendezvousInterface::ParsedKey& key,
-                                     const RendezvousInterface::Args& args,
-                                     const Tensor& tensor, bool is_dead);
+absl::StatusOr<SendParamPtr> SendParamsToC(
+    const RendezvousInterface::ParsedKey& key,
+    const RendezvousInterface::Args& args, const Tensor& tensor, bool is_dead);
 
 void RendezvousCallbackThunk(void* context,
                              TF_RendezvousDoneCallback_Params* params) {
@@ -168,9 +168,10 @@ SendParamDeleter MakeSendParamDeleter() {
   };
 }
 
-StatusOr<SendParamPtr> SendParamsToC(const RendezvousInterface::ParsedKey& key,
-                                     const RendezvousInterface::Args& args,
-                                     const Tensor& tensor, const bool is_dead) {
+absl::StatusOr<SendParamPtr> SendParamsToC(
+    const RendezvousInterface::ParsedKey& key,
+    const RendezvousInterface::Args& args, const Tensor& tensor,
+    const bool is_dead) {
   TF_RendezvousSend_Params* params = new TF_RendezvousSend_Params();
   params->key = new TF_RendezvousParsedKey(ToC(key));
   params->args = new TF_RendezvousArgsStruct(ToC(args));
diff --git a/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_internal.cc b/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_internal.cc
index b63803e3012c15..87b6a7dc8c652e 100644
--- a/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_internal.cc
+++ b/tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_internal.cc
@@ -113,7 +113,7 @@ DoneCallbackParamDeleter MakeDoneCallbackParamDeleter() {
   };
 }
 
-StatusOr<DoneCallbackParamPtr> DoneCallbackParamsToC(
+absl::StatusOr<DoneCallbackParamPtr> DoneCallbackParamsToC(
     const Status& status, const RendezvousInterface::Args& sender_args,
     const RendezvousInterface::Args& recver_args, const Tensor& tensor,
     const bool is_dead) {