Searched refs:replica_groups_ (Results 1 – 2 of 2) sorted by relevance
291 return replica_groups_; in replica_groups()309 std::vector<ReplicaGroup> replica_groups_; variable
480 : HloInstruction(opcode, shape), replica_groups_(replica_groups) { in HloCollectiveInstruction()488 *proto.mutable_replica_groups() = {replica_groups_.begin(), in ToProto()489 replica_groups_.end()}; in ToProto()