Home
last modified time | relevance | path

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

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