Home
last modified time | relevance | path

Searched refs:window_first_wordpiece_index (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/native/annotator/pod_ner/
Dutils.cc102 int window_first_wordpiece_index = std::max( in ExpandWindowAndAlign() local
106 if ((window_first_wordpiece_index + max_num_wordpieces_in_window) > in ExpandWindowAndAlign()
108 window_first_wordpiece_index = in ExpandWindowAndAlign()
112 window_first_wordpiece_index, in ExpandWindowAndAlign()
113 std::min(window_first_wordpiece_index + max_num_wordpieces_in_window, in ExpandWindowAndAlign()
132 int window_first_wordpiece_index = wordpiece_span.begin; in FindFullTokensSpanInWindow() local
134 word_starts, window_first_wordpiece_index); in FindFullTokensSpanInWindow()
135 window_first_wordpiece_index = word_starts[*first_token_index]; in FindFullTokensSpanInWindow()
139 wordpiece_span.end, window_first_wordpiece_index + max_num_wordpieces); in FindFullTokensSpanInWindow()
148 return WordpieceSpan(window_first_wordpiece_index, wordpiece_window_end); in FindFullTokensSpanInWindow()
Dutils_test.cc832 int window_first_wordpiece_index = 10; in TEST() local
839 window_first_wordpiece_index = 10; in TEST()