Searched refs:outer_output_tmp_sharding (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 2467 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in GetDotGroupPartitionContractingOutputShardings() local
2481 outer_output_tmp_sharding = UngroupSharding(grouped); in GetDotGroupPartitionContractingOutputShardings()
2496 outer_output_tmp_sharding = output_sharding; in GetDotGroupPartitionContractingOutputShardings()
2519 outer_output_tmp_sharding = UngroupSharding(grouped); in GetDotGroupPartitionContractingOutputShardings()
2538 return std::make_pair(inner_output_sharding, outer_output_tmp_sharding); in GetDotGroupPartitionContractingOutputShardings()
2658 HloSharding outer_output_tmp_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local
2661 std::tie(inner_output_sharding, outer_output_tmp_sharding) = in PartitionDotGroupOnContracting()
2694 b, result, outer_output_tmp_sharding, lhs.state().next_channel_id, in PartitionDotGroupOnContracting()
2709 outer_output_tmp_sharding, get_non_slice_dims())) in PartitionDotGroupOnContracting()
2715 b, result, outer_output_tmp_sharding, lhs.state().next_channel_id, in PartitionDotGroupOnContracting()
[all …]
|