Searched defs:matching_sharding (Results 1 – 1 of 1) sorted by relevance
2244 bool lhs_matching, const HloSharding& matching_sharding, in GetNonContractingPartitionGroupedShardingForMatchedOperand()2877 const HloSharding& matching_sharding = in EstimateWindowedEinsumIterationsForNonContractingPartitioning() local