Searched defs:num_shard_rows (Results 1 – 1 of 1) sorted by relevance
271 const int64_t row_begin, const int64_t num_shard_rows, in GetSparseMatrixRef()317 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS() local388 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS() local453 const int64_t num_shard_rows = in HandleBatchAndRowRange() local