Home
last modified time | relevance | path

Searched refs:selected_span (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/native/annotator/
Dcached-features.cc97 TokenSpan selected_span, std::vector<float>* output_features) const { in AppendBoundsSensitiveFeaturesForSpan() argument
101 selected_span.first -= extraction_span_.first; in AppendBoundsSensitiveFeaturesForSpan()
102 selected_span.second -= extraction_span_.first; in AppendBoundsSensitiveFeaturesForSpan()
108 /*intended_span=*/{selected_span.first - config->num_tokens_before(), in AppendBoundsSensitiveFeaturesForSpan()
109 selected_span.first + in AppendBoundsSensitiveFeaturesForSpan()
111 /*read_mask_span=*/{0, selected_span.second}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
117 /*intended_span=*/{selected_span.second - in AppendBoundsSensitiveFeaturesForSpan()
119 selected_span.second + config->num_tokens_after()}, in AppendBoundsSensitiveFeaturesForSpan()
121 {selected_span.first, extraction_span_.Size()}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
124 AppendBagFeatures(selected_span, output_features); in AppendBoundsSensitiveFeaturesForSpan()
[all …]
Dcached-features_test.cc59 const CachedFeatures& cached_features, TokenSpan selected_span) { in GetCachedBoundsSensitiveFeatures() argument
61 cached_features.AppendBoundsSensitiveFeaturesForSpan(selected_span, in GetCachedBoundsSensitiveFeatures()
Dcached-features.h47 TokenSpan selected_span, std::vector<float>* output_features) const;