Home
last modified time | relevance | path

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

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