Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h148 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64 num_partitions,
Dspmd_partitioner.cc3411 SPMDCollectiveOpsCreator GetDefaultCollectiveOpsCreator(int64 num_partitions, in GetDefaultCollectiveOpsCreator() function
3497 GetDefaultCollectiveOpsCreator(num_partitions, num_replicas)) {} in SpmdPartitioner()
Dspmd_partitioner_test.cc50 GetDefaultCollectiveOpsCreator(num_devices, /*num_replicas=*/1); in PartitionComputation()