Searched defs:batch_begin (Results 1 – 7 of 7) sorted by relevance
114 [&](int64_t batch_begin, int64_t batch_end) { in Compute()193 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
99 amd_cost_per_batch, [&](int64_t batch_begin, int64_t batch_end) { in Compute()
95 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
99 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
186 matmul_cost_per_batch, [&](int64_t batch_begin, int64_t batch_end) { in Compute()228 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
439 const int64_t batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange() local
155 const int64_t batch_begin = begin / m; in operator ()() local