Searched refs:to_combine (Results 1 – 2 of 2) sorted by relevance
49 Status CombineAllReduces(absl::Span<HloInstruction* const> to_combine) { in CombineAllReduces() argument50 if (to_combine.size() < 2) { in CombineAllReduces()53 VLOG(1) << "Combined " << to_combine.size() << " CRS ops"; in CombineAllReduces()55 HloComputation& computation = *to_combine.back()->parent(); in CombineAllReduces()56 HloComputation* reduction = to_combine[0]->to_apply(); in CombineAllReduces()64 for (HloInstruction* hlo : to_combine) { in CombineAllReduces()84 to_combine.front()->replica_groups(), in CombineAllReduces()85 /*constrain_layout=*/false, to_combine.front()->channel_id(), in CombineAllReduces()86 Cast<HloAllReduceInstruction>(to_combine.front()) in CombineAllReduces()91 if (to_combine.front()->has_sharding()) { in CombineAllReduces()[all …]
171 auto assign_or_combine_shapetree = [&](ShapeTree<bool>&& to_combine, in ComputeHloReplicationOnComputation()175 hlo_replication_[dest] = std::move(to_combine); in ComputeHloReplicationOnComputation()181 if (*element && !to_combine.element(index)) { in ComputeHloReplicationOnComputation()