Automated Code Change

PiperOrigin-RevId: 624401549
tensorflow · Apr 13, 2024 · 7cccea5 · 7cccea5
1 parent 177a5ad
commit 7cccea5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/third_party/xla/xla/service/constant_value.cc b/third_party/xla/xla/service/constant_value.cc
@@ -22,7 +22,7 @@ namespace xla {
 absl::StatusOr<ConstantValue> ConstantValue::FromLiteral(
     const Literal& literal) {
   CHECK_EQ(literal.shape().dimensions_size(), 0) << "Expected scalar literal";
-  return primitive_util::PrimitiveTypeSwitch<StatusOr<ConstantValue>>(
+  return primitive_util::PrimitiveTypeSwitch<absl::StatusOr<ConstantValue>>(
       [&](auto primitive_type_constant) -> absl::StatusOr<ConstantValue> {
         if constexpr (primitive_util::IsIntegralType(primitive_type_constant)) {
           return ConstantValue(

diff --git a/third_party/xla/xla/service/hlo_runner.cc b/third_party/xla/xla/service/hlo_runner.cc
@@ -525,7 +525,7 @@ absl::StatusOr<std::vector<Literal>> HloRunner::ExecuteReplicated(
                                                     argument_buffer_slices));
         } else {
           absl::Mutex mutex;
-          std::vector<StatusOr<ScopedShapedBuffer>> thread_results(
+          std::vector<absl::StatusOr<ScopedShapedBuffer>> thread_results(
               options.num_replicas);
           {
             VLOG(1) << "Creating thread pool for " << options.num_replicas
@@ -581,7 +581,7 @@ absl::StatusOr<std::vector<Literal>> HloRunner::ExecuteReplicated(
         TF_RET_CHECK(options.use_threads);
         std::vector<ScopedShapedBuffer> results;
         absl::Mutex mutex;
-        std::vector<StatusOr<ScopedShapedBuffer>> thread_results(
+        std::vector<absl::StatusOr<ScopedShapedBuffer>> thread_results(
             options.num_replicas);
         {
           VLOG(1) << "Creating thread pool for " << options.num_replicas