Searched refs:k_dim_b (Results 1 – 1 of 1) sorted by relevance
704 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local728 k_dim_b = b_matrix_shape.dim(1).size(); in CountMatMulOperations()731 k_dim_b = b_matrix_shape.dim(0).size(); in CountMatMulOperations()738 if (k_dim_b != 1 && k_dim != 1 && k_dim_b != k_dim) { in CountMatMulOperations()743 k_dim = std::max(k_dim, k_dim_b); in CountMatMulOperations()