No public description

PiperOrigin-RevId: 645935675
tensorflow · Jun 24, 2024 · dd91905 · dd91905
1 parent 272d08b
commit dd91905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_text/core/kernels/sentencepiece/sentencepiece_detokenizer_kernel.cc b/tensorflow_text/core/kernels/sentencepiece/sentencepiece_detokenizer_kernel.cc
@@ -68,7 +68,7 @@ class TFSentencepieceDetokenizerOp : public tensorflow::OpKernel {
       const auto res = sentencepiece::DecodeString(
           codes_for_split, model_tensor.data());
       OP_REQUIRES(ctx, res.type == sentencepiece::DecoderResultType::SUCCESS,
-                  tensorflow::Status(static_cast<tsl::errors::Code>(
+                  tensorflow::Status(static_cast<absl::StatusCode>(
                                          absl::StatusCode::kInternal),
                                      "Sentencepiece conversion failed"));
       output_flat(i) = res.decoded;

diff --git a/tensorflow_text/core/kernels/sentencepiece/sentencepiece_tokenizer_kernel.cc b/tensorflow_text/core/kernels/sentencepiece/sentencepiece_tokenizer_kernel.cc
@@ -65,7 +65,7 @@ class TFSentencepieceOp : public tensorflow::OpKernel {
       const auto res = sentencepiece::EncodeString(
           input_values_flat(i), model_tensor.data(), add_bos, add_eos, reverse);
       OP_REQUIRES(ctx, res.type == sentencepiece::EncoderResultType::SUCCESS,
-                  tensorflow::Status(static_cast<tsl::errors::Code>(
+                  tensorflow::Status(static_cast<absl::StatusCode>(
                                          absl::StatusCode::kInternal),
                                      "Sentencepiece conversion failed"));
       std::copy(res.codes.begin(), res.codes.end(),