Searched refs:create_cross_partition_collective_permute (Results 1 – 5 of 5) sorted by relevance
294 collective_ops_creator.create_cross_partition_collective_permute( in GetFinalFftUsingCollectivePermute()298 collective_ops_creator.create_cross_partition_collective_permute( in GetFinalFftUsingCollectivePermute()
648 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()1166 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()1179 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()1201 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()1214 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()1242 .create_cross_partition_collective_permute( in PartitionBaseCase()1251 .create_cross_partition_collective_permute( in PartitionBaseCase()1282 .create_cross_partition_collective_permute( in PartitionBaseCase()1304 .create_cross_partition_collective_permute( in PartitionBaseCase()1364 .collective_ops_creator.create_cross_partition_collective_permute( in PartitionBaseCase()[all …]
129 create_cross_partition_collective_permute; member
845 collective_ops_creator.create_cross_partition_collective_permute( in ExchangeHalo()878 collective_ops_creator.create_cross_partition_collective_permute( in ExchangeHalo()1627 result.create_cross_partition_collective_permute = in GetPerGroupCollectiveOpsCreator()1642 return creator.create_cross_partition_collective_permute( in GetPerGroupCollectiveOpsCreator()
1345 state_.collective_ops_creator.create_cross_partition_collective_permute( in ReshardWithCollectivePermute()