Searched refs:outer_output_tmp_sharding (Results 1 – 1 of 1) sorted by relevance
2175 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2187 outer_output_tmp_sharding = UngroupSharding(grouped); in PartitionDotGroupOnContracting()2215 outer_output_tmp_sharding = in PartitionDotGroupOnContracting()2241 ar->set_sharding(outer_output_tmp_sharding); in PartitionDotGroupOnContracting()