Home
last modified time | relevance | path

Searched defs:CreateReplicaGroups (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_reduce_combiner_test.cc99 std::vector<ReplicaGroup> CreateReplicaGroups( in CreateReplicaGroups() function
H A Dhlo_parser.cc609 std::vector<ReplicaGroup> CreateReplicaGroups( in CreateReplicaGroups() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2028 std::vector<ReplicaGroup> SpmdPartitioningVisitor::CreateReplicaGroups( in CreateReplicaGroups() function in xla::spmd::SpmdPartitioningVisitor