Searched refs:matching_grouped (Results 1 – 1 of 1) sorted by relevance
2010 GroupedSharding matching_grouped = in PartitionDotGroupOnNonContracting() local2015 matching.sharding() != UngroupSharding(matching_grouped)) { in PartitionDotGroupOnNonContracting()2032 matching = matching.Reshard(UngroupSharding(matching_grouped)); in PartitionDotGroupOnNonContracting()2034 matching.state(), matching_grouped.device_groups, b); in PartitionDotGroupOnNonContracting()2036 matching.hlo()->set_sharding(matching_grouped.sharding); in PartitionDotGroupOnNonContracting()2039 GetPerGroupBaseShape(matching_grouped, matching.base_shape()), in PartitionDotGroupOnNonContracting()2069 num_partitions / matching_grouped.device_groups.size(), in PartitionDotGroupOnNonContracting()2373 GroupedSharding matching_grouped = in LhsIsBestMatchForNonContractingPartitioning() local2412 matching_grouped.sharding, other_grouped->sharding); in LhsIsBestMatchForNonContractingPartitioning()