Searched refs:VectorSpan (Results 1 – 8 of 8) sorted by relevance
| /external/libtextclassifier/native/annotator/pod_ner/ |
| D | utils_test.cc | 121 VectorSpan<Token>(TokenizeOnSpace(text)), tags, in TEST_P()
128 VectorSpan<Token>(TokenizeOnSpace(text)), TagsToLabels(tags), in TEST_P()
147 VectorSpan<Token>(TokenizeOnSpace(text)), tags, in TEST_P()
154 VectorSpan<Token>(TokenizeOnSpace(text)), TagsToLabels(tags), in TEST_P()
185 VectorSpan<Token>(TokenizeOnSpace(text)), tags, in TEST_P()
192 VectorSpan<Token>(TokenizeOnSpace(text)), TagsToLabels(tags), in TEST_P()
231 VectorSpan<Token>(original_tokens.begin() + 2, original_tokens.end()), in TEST_P()
239 VectorSpan<Token>(original_tokens.begin() + 2, original_tokens.end()), in TEST_P()
264 VectorSpan<Token>(TokenizeOnSpace(text)), TagsToLabels(tags), in TEST()
283 VectorSpan<Token>(TokenizeOnSpace(text)), tags, in TEST_P()
[all …]
|
| D | utils.h | 85 bool ConvertTagsToAnnotatedSpans(const VectorSpan<Token> &tokens,
100 const VectorSpan<Token> &tokens,
126 bool Next(VectorSpan<int32_t> *cur_wordpiece_indices,
127 VectorSpan<int32_t> *cur_token_starts,
128 VectorSpan<Token> *cur_tokens);
|
| D | utils.cc | 174 bool WindowGenerator::Next(VectorSpan<int32_t> *cur_wordpiece_indices, in Next()
175 VectorSpan<int32_t> *cur_token_starts, in Next()
176 VectorSpan<Token> *cur_tokens) { in Next()
185 *cur_token_starts = VectorSpan<int32_t>( in Next()
188 *cur_tokens = VectorSpan<Token>( in Next()
208 *cur_wordpiece_indices = VectorSpan<int>( in Next()
216 bool ConvertTagsToAnnotatedSpans(const VectorSpan<Token> &tokens, in ConvertTagsToAnnotatedSpans()
326 bool ConvertTagsToAnnotatedSpans(const VectorSpan<Token> &tokens, in ConvertTagsToAnnotatedSpans()
|
| D | pod-ner-impl.h | 61 const VectorSpan<int> &wordpiece_indices,
62 const VectorSpan<int32_t> &token_starts,
63 const VectorSpan<Token> &tokens) const;
|
| D | pod-ner-impl.cc | 270 const VectorSpan<int> &wordpiece_indices, in ExecuteModel()
271 const VectorSpan<int32_t> &token_starts, in ExecuteModel()
272 const VectorSpan<Token> &tokens) const { in ExecuteModel()
429 VectorSpan<int32_t> cur_wordpiece_indices; in AnnotateAroundSpanOfInterest()
430 VectorSpan<int32_t> cur_token_starts; in AnnotateAroundSpanOfInterest()
431 VectorSpan<Token> cur_tokens; in AnnotateAroundSpanOfInterest()
456 VectorSpan<Token>(tokens.begin() + first_token_index_entire_window, in AnnotateAroundSpanOfInterest()
|
| /external/libtextclassifier/native/annotator/ |
| D | feature-processor.h | 180 VectorSpan<Token> tokens,
239 bool LabelToSpan(int label, const VectorSpan<Token>& output_tokens,
|
| D | types.h | 717 class VectorSpan {
719 VectorSpan() : begin_(), end_() {} in VectorSpan() function
720 explicit VectorSpan(const std::vector<T>& v) // NOLINT(runtime/explicit) in VectorSpan() function
722 VectorSpan(typename std::vector<T>::const_iterator begin, in VectorSpan() function
|
| D | feature-processor.cc | 199 const VectorSpan<Token>& tokens, in LabelToSpan()
474 const VectorSpan<Token> tokens, in SelectionLabelSpans()
|