Lines Matching refs:selected_span
98 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()
130 static_cast<float>(TokenSpanSize(selected_span))); in AppendBoundsSensitiveFeaturesForSpan()