Searched refs:output_dimension_number (Results 1 – 1 of 1) sorted by relevance
227 auto output_dimension_number = [&](char d) { in Einsum() local234 output_dims.push_back(output_dimension_number(x_config[d])); in Einsum()237 output_dims.push_back(output_dimension_number(x_config[d])); in Einsum()240 output_dims.push_back(output_dimension_number(y_config[d])); in Einsum()