Home
last modified time | relevance | path

Searched refs:create_cross_partition_all_to_all (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc165 auto all_to_all = collective_ops_creator.create_cross_partition_all_to_all( in ShuffleDataWithAllToAll()
H A Dspmd_partitioner.h148 create_cross_partition_all_to_all; member
H A Dspmd_partitioner_util.cc1531 result.create_cross_partition_all_to_all = in GetPerGroupCollectiveOpsCreator()
1536 return creator.create_cross_partition_all_to_all( in GetPerGroupCollectiveOpsCreator()
H A Dspmd_partitioner.cc1586 state_.collective_ops_creator.create_cross_partition_all_to_all( in ReshardWithAllToAll()