Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc284 std::vector<std::pair<int64, int64>> src_dst_pairs; in GetFinalFftUsingCollectivePermute() local
Dspmd_partitioner_util.cc1678 int64_t next_channel_id) { in GetPerGroupCollectiveOpsCreator()
Dspmd_partitioner.cc1388 std::vector<std::pair<int64, int64>> src_dst_pairs; in ReshardWithCollectivePermute() local