Searched refs:outer_output_tmp_sharding (Results 1 – 1 of 1) sorted by relevance
2175 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in GetDotGroupPartitionContractingOutputShardings() local2188 outer_output_tmp_sharding = UngroupSharding(grouped); in GetDotGroupPartitionContractingOutputShardings()2215 outer_output_tmp_sharding = UngroupSharding(grouped); in GetDotGroupPartitionContractingOutputShardings()2221 return std::make_pair(inner_output_sharding, outer_output_tmp_sharding); in GetDotGroupPartitionContractingOutputShardings()2340 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local2342 std::tie(inner_output_sharding, outer_output_tmp_sharding) = in PartitionDotGroupOnContracting()2398 outer_output_tmp_sharding = in PartitionDotGroupOnContracting()2400 outer_output_tmp_sharding, output_slice_dims); in PartitionDotGroupOnContracting()2429 dot->set_sharding(outer_output_tmp_sharding); in PartitionDotGroupOnContracting()2691 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local[all …]