Searched refs:output_transpose_dims (Results 1 – 1 of 1) sorted by relevance
452 absl::InlinedVector<int64, 8> output_transpose_dims; in Einsum() local463 output_transpose_dims.push_back(*output_dimension_number(x_config[d])); in Einsum()468 output_transpose_dims.push_back(*output_dim); in Einsum()476 output_transpose_dims.push_back(*output_dim); in Einsum()482 const int64 transpose_rank = output_transpose_dims.size(); in Einsum()485 transpose_dims[output_transpose_dims[i]] = i; in Einsum()