Searched defs:num_rhs_rows (Results 1 – 1 of 1) sorted by relevance
182 int64_t num_rhs_rows = matrix_b.dim_size(rank - 2); in Compute() local299 const int64_t num_rhs_rows = rhs.dim_size(rhs.dims() - 2); in SparseDenseMatMulWithoutTransposedLHS() local347 const int64_t num_rhs_rows = rhs.dim_size(rhs.dims() - 2); in SparseDenseMatMulWithTransposedLHS() local