Searched refs:cur_wordpiece_indices (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/native/annotator/pod_ner/ |
| D | utils_test.cc | 856 VectorSpan<int32_t> cur_wordpiece_indices; in TEST() local
859 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
862 ASSERT_EQ(cur_wordpiece_indices.size(), 3); in TEST()
864 ASSERT_EQ(cur_wordpiece_indices[i], wordpiece_indices[i]); in TEST()
873 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
876 ASSERT_EQ(cur_wordpiece_indices.size(), 4); in TEST()
877 for (int i = 0; i < cur_wordpiece_indices.size(); i++) { in TEST()
878 ASSERT_EQ(cur_wordpiece_indices[i], wordpiece_indices[i + 2]); in TEST()
887 ASSERT_TRUE(window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in TEST()
890 ASSERT_EQ(cur_wordpiece_indices.size(), 3); in TEST()
[all …]
|
| D | pod-ner-impl.cc | 425 VectorSpan<int32_t> cur_wordpiece_indices; in AnnotateAroundSpanOfInterest() local
428 if (!window_generator.Next(&cur_wordpiece_indices, &cur_token_starts, in AnnotateAroundSpanOfInterest()
431 cur_wordpiece_indices.size() <= 0) { in AnnotateAroundSpanOfInterest()
435 ExecuteModel(cur_wordpiece_indices, cur_token_starts, cur_tokens); in AnnotateAroundSpanOfInterest()
|
| D | utils.h | 126 bool Next(VectorSpan<int32_t> *cur_wordpiece_indices,
|
| D | utils.cc | 174 bool WindowGenerator::Next(VectorSpan<int32_t> *cur_wordpiece_indices, in Next() argument
208 *cur_wordpiece_indices = VectorSpan<int>( in Next()
|