Searched defs:num_shard_rows (Results 1 – 1 of 1) sorted by relevance
265 const int64_t row_begin, const int64_t num_shard_rows, in GetSparseMatrixRef()311 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS() local382 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS() local447 const int64_t num_shard_rows = in HandleBatchAndRowRange() local