Searched defs:outer_output_tmp_sharding (Results 1 – 1 of 1) sorted by relevance
2467 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in GetDotGroupPartitionContractingOutputShardings() local2658 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local3069 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local