Searched refs:k_dim_b (Results 1 – 1 of 1) sorted by relevance
1057 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local1081 k_dim_b = b_matrix_shape.dim(1).size(); in CountMatMulOperations()1084 k_dim_b = b_matrix_shape.dim(0).size(); in CountMatMulOperations()1091 if (k_dim_b != 1 && k_dim != 1 && k_dim_b != k_dim) { in CountMatMulOperations()1096 k_dim = std::max(k_dim, k_dim_b); in CountMatMulOperations()