Searched refs:create_cross_partition_all_to_all (Results 1 – 4 of 4) sorted by relevance
167 auto all_to_all = collective_ops_creator.create_cross_partition_all_to_all( in ShuffleDataWithAllToAll()
136 create_cross_partition_all_to_all; member
1645 result.create_cross_partition_all_to_all = in GetPerGroupCollectiveOpsCreator()1650 return creator.create_cross_partition_all_to_all( in GetPerGroupCollectiveOpsCreator()
1185 state_.collective_ops_creator.create_cross_partition_all_to_all( in ReshardWithAllToAll()