Automated Code Change

PiperOrigin-RevId: 632099660
tensorflow · May 9, 2024 · 71ef21e · 71ef21e
1 parent cff4685
commit 71ef21e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/dtensor/mlir/restore_shape_inference.cc b/tensorflow/dtensor/mlir/restore_shape_inference.cc
@@ -129,7 +129,7 @@ mlir::LogicalResult BackwardShapeInferenceToRestoreOp(mlir::ModuleOp module,
         module, new_recv_op);
 
     if (!send_op.ok())
-      return recv_op.emitOpError(tsl::NullTerminatedMessage(send_op.status()));
+      return recv_op.emitOpError(absl::StatusMessageAsCStr(send_op.status()));
 
     // Recursively shape inference to the input of the send op.
     return BackwardShapeInferenceToRestoreOp(