Searched refs:output_dimension_number (Results 1 – 1 of 1) sorted by relevance
454 auto output_dimension_number = [&](int64 d) -> absl::optional<int64> { in Einsum() local463 output_transpose_dims.push_back(*output_dimension_number(x_config[d])); in Einsum()467 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum()475 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum()