Searched refs:dim_name (Results 1 – 1 of 1) sorted by relevance
207 auto dim_name = x_config[i]; in Einsum() local208 if (is_batch_dim(dim_name)) { in Einsum()210 dnums.add_rhs_batch_dimensions(rhs_dimension_number(dim_name)); in Einsum()211 } else if (is_contracting(dim_name)) { in Einsum()213 dnums.add_rhs_contracting_dimensions(rhs_dimension_number(dim_name)); in Einsum()221 auto dim_name = y_config[i]; in Einsum() local222 if (!is_batch_dim(dim_name) && !is_contracting(dim_name)) { in Einsum()