Searched defs:row_begin (Results 1 – 4 of 4) sorted by relevance
265 const int64_t row_begin, const int64_t num_shard_rows, in GetSparseMatrixRef()310 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithoutTransposedLHS()381 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithTransposedLHS()446 const int64_t row_begin = current_batch_row_begin % num_rows; in HandleBatchAndRowRange() local
238 const int64_t row_begin = input.row_ptr(row_idx); in operator ()() local
158 const int64_t row_begin = begin > batch * m ? begin % m : 0; in operator ()() local
176 const int64_t row_begin = begin[0]; in Compute() local