Searched defs:inner_output_sharding (Results 1 – 1 of 1) sorted by relevance
2174 HloSharding inner_output_sharding = HloSharding::Replicate(); in GetDotGroupPartitionContractingOutputShardings() local2339 HloSharding inner_output_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2690 HloSharding inner_output_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local