Skip to content

Commit a572eaf

Browse files
No public description
PiperOrigin-RevId: 795371275
1 parent b74a866 commit a572eaf

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

tensorflow_text/core/kernels/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1018,6 +1018,7 @@ tf_cc_library(
10181018
],
10191019
deps = [
10201020
"@com_google_absl//absl/base:core_headers",
1021+
"@com_google_absl//absl/status",
10211022
"@com_google_absl//absl/strings",
10221023
"@icu//:common",
10231024
],

tensorflow_text/core/kernels/sentencepiece_kernels.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#include "absl/base/thread_annotations.h"
1818
#include "absl/container/flat_hash_map.h"
1919
#include "absl/meta/type_traits.h"
20+
#include "absl/status/status.h"
2021
#include "absl/strings/string_view.h"
2122
#include "absl/synchronization/mutex.h"
2223
#include "absl/types/span.h"
@@ -95,7 +96,7 @@ constexpr int64 kCostPerUnit = 10000;
9596

9697
::tensorflow::Status ToTFStatus(const sentencepiece::util::Status& s) {
9798
if (s.ok()) return ::tensorflow::Status();
98-
return ::tensorflow::Status(static_cast<::tensorflow::errors::Code>(s.code()),
99+
return ::tensorflow::Status(static_cast<::absl::StatusCode>(s.code()),
99100
::tensorflow::string(s.message()));
100101
}
101102

tensorflow_text/core/kernels/split_merge_tokenize_kernel.cc

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#include <string>
1818
#include <vector>
1919

20+
#include "absl/status/status.h"
2021
#include "absl/strings/str_cat.h"
2122
#include "icu4c/source/common/unicode/uchar.h"
2223
#include "icu4c/source/common/unicode/umachine.h"
@@ -72,13 +73,13 @@ Status TokenizeByLabel(const absl::string_view& text,
7273
std::vector<int>* end_offset, int* num_tokens) {
7374
std::vector<absl::string_view> chars;
7475
if (!GetUTF8Chars(text, &chars)) {
75-
return Status(static_cast<tensorflow::errors::Code>(
76+
return Status(static_cast<::absl::StatusCode>(
7677
absl::StatusCode::kInvalidArgument),
7778
absl::StrCat("Input string is not utf8 valid: ", text));
7879
}
7980

8081
if (chars.size() > labels_tensor.dim_size(0)) {
81-
return Status(static_cast<tensorflow::errors::Code>(
82+
return Status(static_cast<::absl::StatusCode>(
8283
absl::StatusCode::kInvalidArgument),
8384
absl::StrCat("Number of labels ", labels_tensor.dim_size(0),
8485
" is insufficient for text ", text));

0 commit comments

Comments
 (0)