Home
last modified time | relevance | path

Searched refs:cur_token_starts (Results 1 – 4 of 4) sorted by relevance

/external/libtextclassifier/native/annotator/pod_ner/
Dutils_test.cc857 VectorSpan<int32_t> cur_token_starts; in TEST() local
859 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
866 ASSERT_EQ(cur_token_starts.size(), 2); in TEST()
869 ASSERT_EQ(cur_token_starts[i], token_starts[i]); in TEST()
873 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
880 ASSERT_EQ(cur_token_starts.size(), 3); in TEST()
883 ASSERT_EQ(cur_token_starts[i], token_starts[i + 1]); in TEST()
887 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
894 ASSERT_EQ(cur_token_starts.size(), 3); in TEST()
897 ASSERT_EQ(cur_token_starts[i], token_starts[i + 3]); in TEST()
[all …]
Dpod-ner-impl.cc426 VectorSpan<int32_t> cur_token_starts; in AnnotateAroundSpanOfInterest() local
428 if (!window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in AnnotateAroundSpanOfInterest()
430 cur_tokens.size() <= 0 || cur_token_starts.size() <= 0 || in AnnotateAroundSpanOfInterest()
435 ExecuteModel(cur_wordpiece_indices, cur_token_starts, cur_tokens); in AnnotateAroundSpanOfInterest()
Dutils.h127 VectorSpan<int32_t> *cur_token_starts,
Dutils.cc175 VectorSpan<int32_t> *cur_token_starts, in Next() argument
185 *cur_token_starts = VectorSpan<int32_t>( in Next()