Searched defs:num_shard_rows (Results 1 – 1 of 1) sorted by relevance
265 const int64 row_begin, const int64 num_shard_rows, in GetSparseMatrixRef()308 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS() local379 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS() local444 const int64 num_shard_rows = in HandleBatchAndRowRange() local