Searched refs:replica_groups_ (Results 1 – 2 of 2) sorted by relevance
350 return replica_groups_; in replica_groups()384 std::vector<ReplicaGroup> replica_groups_; variable
576 replica_groups_(replica_groups), in HloCollectiveInstruction()585 *proto.mutable_replica_groups() = {replica_groups_.begin(), in ToProto()586 replica_groups_.end()}; in ToProto()