Automated Code Change

PiperOrigin-RevId: 609306481
tensorflow · Feb 22, 2024 · c9ca63d · c9ca63d
1 parent 2a09127
commit c9ca63d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 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
@@ -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
@@ -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) {