Lines Matching refs:batch_start
2168 for (int batch_start = span_of_interest.first; in ModelClickContextScoreChunks() local
2169 batch_start < span_of_interest.second; batch_start += max_batch_size) { in ModelClickContextScoreChunks()
2171 std::min(batch_start + max_batch_size, span_of_interest.second); in ModelClickContextScoreChunks()
2176 for (int click_pos = batch_start; click_pos < batch_end; ++click_pos) { in ModelClickContextScoreChunks()
2182 const int batch_size = batch_end - batch_start; in ModelClickContextScoreChunks()
2199 for (int click_pos = batch_start; click_pos < batch_end; ++click_pos) { in ModelClickContextScoreChunks()
2201 logits.data() + logits.dim(1) * (click_pos - batch_start), in ModelClickContextScoreChunks()
2279 for (int batch_start = 0; batch_start < candidate_spans.size(); in ModelBoundsSensitiveScoreChunks() local
2280 batch_start += max_batch_size) { in ModelBoundsSensitiveScoreChunks()
2281 const int batch_end = std::min(batch_start + max_batch_size, in ModelBoundsSensitiveScoreChunks()
2287 for (int i = batch_start; i < batch_end; ++i) { in ModelBoundsSensitiveScoreChunks()
2293 const int batch_size = batch_end - batch_start; in ModelBoundsSensitiveScoreChunks()
2309 for (int i = batch_start; i < batch_end; ++i) { in ModelBoundsSensitiveScoreChunks()
2311 ScoredChunk{candidate_spans[i], logits.data()[i - batch_start]}); in ModelBoundsSensitiveScoreChunks()