Home
last modified time | relevance | path

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

/external/libtextclassifier/annotator/
Dcached-features.cc98 TokenSpan selected_span, std::vector<float>* output_features) const { in AppendBoundsSensitiveFeaturesForSpan() argument
102 selected_span.first -= extraction_span_.first; in AppendBoundsSensitiveFeaturesForSpan()
103 selected_span.second -= extraction_span_.first; in AppendBoundsSensitiveFeaturesForSpan()
109 /*intended_span=*/{selected_span.first - config->num_tokens_before(), in AppendBoundsSensitiveFeaturesForSpan()
110 selected_span.first + in AppendBoundsSensitiveFeaturesForSpan()
112 /*read_mask_span=*/{0, selected_span.second}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
118 /*intended_span=*/{selected_span.second - in AppendBoundsSensitiveFeaturesForSpan()
120 selected_span.second + config->num_tokens_after()}, in AppendBoundsSensitiveFeaturesForSpan()
121 /*read_mask_span=*/{selected_span.first, TokenSpanSize(extraction_span_)}, in AppendBoundsSensitiveFeaturesForSpan()
125 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;