Searched refs:GetDefaultCollectiveOpsCreator (Results 1 – 3 of 3) sorted by relevance
160 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64_t num_partitions,
4220 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64_t num_partitions, in GetDefaultCollectiveOpsCreator() function4325 GetDefaultCollectiveOpsCreator(num_partitions, num_replicas)) {} in SpmdPartitioner()
61 GetDefaultCollectiveOpsCreator(num_devices, /*num_replicas=*/1); in PartitionComputation()