Lines Matching refs:batch_start
2971 for (int batch_start = span_of_interest.first; in ModelClickContextScoreChunks() local
2972 batch_start < span_of_interest.second; batch_start += max_batch_size) { in ModelClickContextScoreChunks()
2974 std::min(batch_start + max_batch_size, span_of_interest.second); in ModelClickContextScoreChunks()
2979 for (int click_pos = batch_start; click_pos < batch_end; ++click_pos) { in ModelClickContextScoreChunks()
2985 const int batch_size = batch_end - batch_start; in ModelClickContextScoreChunks()
3002 for (int click_pos = batch_start; click_pos < batch_end; ++click_pos) { in ModelClickContextScoreChunks()
3004 logits.data() + logits.dim(1) * (click_pos - batch_start), in ModelClickContextScoreChunks()
3080 for (int batch_start = 0; batch_start < candidate_spans.size(); in ModelBoundsSensitiveScoreChunks() local
3081 batch_start += max_batch_size) { in ModelBoundsSensitiveScoreChunks()
3082 const int batch_end = std::min(batch_start + max_batch_size, in ModelBoundsSensitiveScoreChunks()
3088 for (int i = batch_start; i < batch_end; ++i) { in ModelBoundsSensitiveScoreChunks()
3094 const int batch_size = batch_end - batch_start; in ModelBoundsSensitiveScoreChunks()
3110 for (int i = batch_start; i < batch_end; ++i) { in ModelBoundsSensitiveScoreChunks()
3112 ScoredChunk{candidate_spans[i], logits.data()[i - batch_start]}); in ModelBoundsSensitiveScoreChunks()