| 616 | } |
| 617 | |
| 618 | std::vector<HloComputation*> HloModule::MakeNonfusionComputations() const { |
| 619 | std::vector<HloComputation*> result; |
| 620 | for (auto* c : computations()) { |
| 621 | if (c->IsFusionComputation()) { |
| 622 | continue; |
| 623 | } |
| 624 | result.push_back(c); |
| 625 | } |
| 626 | return result; |
| 627 | } |
| 628 | |
| 629 | std::vector<HloComputation*> HloModule::MakeNonfusionComputationsSorted() |
| 630 | const { |