Automated Code Change

PiperOrigin-RevId: 660273619

Author: tfx-copybara

struct2tensor/kernels/decode_proto_map_op.cc

@@ -84,27 +84,27 @@ Status ParseStringAs(const std::string& str, T* val) {
     return errors::InvalidArgument(
         absl::StrCat("Failed to parse string: ", str, " as integer."));
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 // Specialization for parsing into a string_view.
 template <>
 Status ParseStringAs<absl::string_view>(const std::string& str,
                                         absl::string_view* val) {
   *val = str;
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 // Specialization for parsing into a boolean.
 template <>
 Status ParseStringAs<bool>(const std::string& str, bool* val) {
   if (str == "0") {
     *val = false;
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
   if (str == "1") {
     *val = true;
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
   return errors::InvalidArgument(
       absl::StrCat("Failed to parse string: ", str, " as bool."));
@@ -225,7 +225,7 @@ class KeyDecoder : public KeyDecoderBase {
     }
 
     *key_decoder = absl::WrapUnique(new KeyDecoder(keys_as_strings));
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status Decode(StreamingProtoReader* reader, int* value_index) const override {
@@ -239,7 +239,7 @@ class KeyDecoder : public KeyDecoderBase {
     } else {
       *value_index = it->second;
     }
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
  private:
@@ -293,7 +293,7 @@ class ValueCollector : public ValueCollectorBase {
     if (!reader->ReadValue(kFieldType, &current_value_)) {
       return errors::DataLoss("Corrupted value field.");
     }
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
   void Commit(const int key_index, const int64_t parent_index) override {
     values_per_key_[key_index].push_back(current_value_);
@@ -393,7 +393,7 @@ class MapEntryCollector {
     }
     *map_entry_collector = absl::WrapUnique(new MapEntryCollector(
         keys_as_strings.size(), std::move(key_decoder), value_type));
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   ~MapEntryCollector() {}
@@ -542,7 +542,7 @@ class MapEntryCollector {
         return errors::InvalidArgument(
             absl::StrCat("Unexpected map value type: ", value_type_));
     }
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status PopulateOutputTensors(const ValueCollectorBase& value_collector,
@@ -567,7 +567,7 @@ class MapEntryCollector {
           i, output_parent_indices_tensor);
     }
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   const int num_keys_;

struct2tensor/kernels/decode_proto_sparse_op.cc

@@ -100,7 +100,7 @@ ::tensorflow::Status ToOutputTensor(OpKernelContext* context,
   if (tensor_size > 0) {
     VectorToTensor(vec, result, produce_string_view);
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 // Makes `value` refer to bytes for a length-delimited field in the buffer
@@ -318,7 +318,7 @@ class FieldBuilderImpl : public FieldBuilder {
         is_packed_primitive = true;
       } else if (WireFormatLite::SkipField(
                      input, WireFormatLite::MakeTag(wire_number_, wire_type))) {
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       } else {
         return DataLoss("Failed skipping malformed field");
       }
@@ -337,7 +337,7 @@ class FieldBuilderImpl : public FieldBuilder {
                                       produce_string_view));
     TF_RETURN_IF_ERROR(ToOutputTensor(context, output_index_parent_index_,
                                       parent_indices_, produce_string_view));
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   void MaybePadDefaultValue(int64_t current_message_index) {
@@ -369,7 +369,7 @@ class FieldBuilderImpl : public FieldBuilder {
       TF_RETURN_IF_ERROR(CollectValue(input, message_index));
     }
     input->PopLimit(limit);
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   // Parses one value from `input`, then updates `values_` and
@@ -388,7 +388,7 @@ class FieldBuilderImpl : public FieldBuilder {
     } else {
       values_.back() = value;
     }
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   // Collected field values.
@@ -1165,7 +1165,7 @@ class DecodeProtoSparseOp : public OpKernel {
           "it.");
     }
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   std::string message_type_;

struct2tensor/kernels/equi_join_any_indices_op.cc

@@ -46,7 +46,7 @@ ::tensorflow::Status ToOutputVector(OpKernelContext* context, int index,
     memcpy(result_flat.data(), vec.data(),
            tensor_size * sizeof(tensorflow::int64));
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace

struct2tensor/kernels/equi_join_indices_op.cc

@@ -50,7 +50,7 @@ ::tensorflow::Status ToOutputVector(OpKernelContext* context, int index,
     memcpy(result_flat.data(), vec.data(),
            tensor_size * sizeof(tensorflow::int64));
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace