Searched defs:num_rhs_cols (Results 1 – 1 of 1) sorted by relevance
189 int64_t num_rhs_cols = matrix_b.dim_size(rank - 1); in Compute() local306 const int64_t num_rhs_cols = rhs.dim_size(rhs.dims() - 1); in SparseDenseMatMulWithoutTransposedLHS() local354 const int64_t num_rhs_cols = rhs.dim_size(rhs.dims() - 1); in SparseDenseMatMulWithTransposedLHS() local