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