Searched refs:num_rhs_rows (Results 1 – 1 of 1) sorted by relevance
182 int64 num_rhs_rows = matrix_b.dim_size(rank - 2); in Compute() local197 std::swap(num_rhs_rows, num_rhs_cols); in Compute()296 const int64 num_rhs_rows = rhs.dim_size(rhs.dims() - 2); in SparseDenseMatMulWithoutTransposedLHS() local318 num_rhs_rows * in SparseDenseMatMulWithoutTransposedLHS()320 num_rhs_rows, num_rhs_cols); in SparseDenseMatMulWithoutTransposedLHS()344 const int64 num_rhs_rows = rhs.dim_size(rhs.dims() - 2); in SparseDenseMatMulWithTransposedLHS() local391 batch_idx * num_rhs_rows * num_rhs_cols + in SparseDenseMatMulWithTransposedLHS()