Searched defs:row_begin (Results 1 – 4 of 4) sorted by relevance
271 const int64_t row_begin, const int64_t num_shard_rows, in GetSparseMatrixRef()316 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithoutTransposedLHS()387 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithTransposedLHS()452 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