Home
last modified time | relevance | path

Searched defs:src_dst_pairs (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc282 std::vector<std::pair<int64_t, int64_t>> src_dst_pairs; in GetFinalFftUsingCollectivePermute() local
H A Dspmd_partitioner_util.cc1517 int64_t next_channel_id) { in GetPerGroupCollectiveOpsCreator()
H A Dspmd_partitioner.cc1975 std::vector<std::pair<int64_t, int64_t>> src_dst_pairs; in ReshardWithCollectivePermute() local
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_inductor_collectives.py267 def func(tensor, src_dst_pairs, *, tag, ranks, group_size): argument