No public description

PiperOrigin-RevId: 646431002
tensorflow · Jun 25, 2024 · 767dfc8 · 767dfc8
1 parent dd91905
commit 767dfc8
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/tensorflow_text/core/kernels/sentencepiece_kernels.cc b/tensorflow_text/core/kernels/sentencepiece_kernels.cc
@@ -83,7 +83,7 @@ struct SentencepieceResource : public ResourceBase {
             .WithName(unique_node_name)
             .WithAttr("model", model)
             .WithAttr("use_node_name_sharing", true));
-    return OkStatus();
+    return absl::OkStatus();
   }
 };
 
@@ -133,13 +133,13 @@ tensorflow::Status HandleExtraOptions(OpKernelContext* ctx,
     // the reader lock once and do a quick check first.
     absl::ReaderMutexLock l(&sp->mu);
     if (sp->SameOptions(add_bos, add_eos, reverse)) {
-      return OkStatus();
+      return absl::OkStatus();
     }
   }
 
   absl::WriterMutexLock lock(&sp->mu);
   if (sp->SameOptions(add_bos, add_eos, reverse)) {
-    return OkStatus();
+    return absl::OkStatus();
   }
   string options;
   sp->add_bos = add_bos;
@@ -164,7 +164,7 @@ tensorflow::Status HandleExtraOptions(OpKernelContext* ctx,
   TF_RETURN_IF_ERROR(ToTFStatus(sp->processor.SetEncodeExtraOptions(options)));
   TF_RETURN_IF_ERROR(ToTFStatus(sp->processor.SetDecodeExtraOptions(options)));
 
-  return OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace
@@ -227,7 +227,7 @@ class SentencepieceOp : public OpKernel {
               }
 
               *resource = sp;
-              return OkStatus();
+              return absl::OkStatus();
             };
 
     // Register the ResourceType alias.

diff --git a/tensorflow_text/core/kernels/split_merge_tokenize_kernel.cc b/tensorflow_text/core/kernels/split_merge_tokenize_kernel.cc
@@ -110,7 +110,7 @@ Status TokenizeByLabel(const absl::string_view& text,
     last_character_is_break_character = is_break_character;
   }
 
-  return OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace

diff --git a/tensorflow_text/core/kernels/tokenizer_from_logits_kernel.cc b/tensorflow_text/core/kernels/tokenizer_from_logits_kernel.cc
@@ -78,13 +78,13 @@ Status TokenizeByLogits(const absl::string_view& text,
   std::vector<absl::string_view> chars;
   if (!GetUTF8Chars(text, &chars)) {
     return Status(
-        static_cast<tsl::errors::Code>(absl::StatusCode::kInvalidArgument),
+        static_cast<absl::StatusCode>(absl::StatusCode::kInvalidArgument),
         absl::StrCat("Input string is not utf8 valid: ", text));
   }
 
   if (chars.size() > logits.dimension(1)) {
     return Status(
-        static_cast<tsl::errors::Code>(absl::StatusCode::kInvalidArgument),
+        static_cast<absl::StatusCode>(absl::StatusCode::kInvalidArgument),
         absl::StrCat("Number of logits, ", logits.dimension(1),
                      ", is insufficient for text \"", text, "\""));
   }
@@ -116,7 +116,7 @@ Status TokenizeByLogits(const absl::string_view& text,
     last_character_is_break_character = is_break_character;
   }
 
-  return OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace

diff --git a/tensorflow_text/core/kernels/wordpiece_kernel.cc b/tensorflow_text/core/kernels/wordpiece_kernel.cc
@@ -99,7 +99,7 @@ Status GetTableHandle(const string& input_name, OpKernelContext* ctx,
     *container = h(0);
     *table_handle = h(1);
   }
-  return OkStatus();
+  return absl::OkStatus();
 }
 
 // Gets the LookupTable stored in the ctx->resource_manager() with key
@@ -137,7 +137,7 @@ class LookupTableVocab : public WordpieceVocab {
 
 Status ToStatus(const LookupStatus& status) {
   if (status.success) {
-    return OkStatus();
+    return absl::OkStatus();
   }
 
   return errors::InvalidArgument(status.error_msg);