Searched refs:output_to_matching_indices (Results 1 – 1 of 1) sorted by relevance
2865 const std::vector<int64_t>& output_to_matching_indices = in EstimateWindowedEinsumIterationsForNonContractingPartitioning() local2933 output_grouped.sharding, output_to_matching_indices, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()