Home
last modified time | relevance | path

Searched refs:GetDefaultCollectiveOpsCreator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.h160 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64_t num_partitions,
H A Dspmd_partitioner.cc4220 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64_t num_partitions, in GetDefaultCollectiveOpsCreator() function
4325 GetDefaultCollectiveOpsCreator(num_partitions, num_replicas)) {} in SpmdPartitioner()
H A Dspmd_partitioner_test.cc61 GetDefaultCollectiveOpsCreator(num_devices, /*num_replicas=*/1); in PartitionComputation()