Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h144 create_cross_partition_all_gather; member
Dspmd_partitioner_util.cc1654 if (creator.create_cross_partition_all_gather) { in GetPerGroupCollectiveOpsCreator()
1655 result.create_cross_partition_all_gather = in GetPerGroupCollectiveOpsCreator()
1660 return creator.create_cross_partition_all_gather( in GetPerGroupCollectiveOpsCreator()
Dspmd_partitioner.cc878 if (state_.collective_ops_creator.create_cross_partition_all_gather) { in ReplicatePartial()
3533 result = collectives_creator.create_cross_partition_all_gather( in AllGatherShardsInternal()
3543 result = collectives_creator.create_cross_partition_all_gather( in AllGatherShardsInternal()
Dspmd_partitioner_test.cc53 collective_ops_creator.create_cross_partition_all_gather = nullptr; in PartitionComputation()