Searched defs:inner_output_sharding (Results 1 – 1 of 1) sorted by relevance
2466 HloSharding inner_output_sharding = HloSharding::Replicate(); in GetDotGroupPartitionContractingOutputShardings() local2657 HloSharding inner_output_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local3068 HloSharding inner_output_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local