MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MakeNonfusionComputations

Method MakeNonfusionComputations

tensorflow/compiler/xla/service/hlo_module.cc:618–627  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

616}
617
618std::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
629std::vector<HloComputation*> HloModule::MakeNonfusionComputationsSorted()
630 const {

Callers 15

RunMethod · 0.80
RunMethod · 0.80
UpdateMethod · 0.80
VerifyMethod · 0.80
RunMethod · 0.80
RunMethod · 0.80
RunMethod · 0.80
RunMethod · 0.80
RunMethod · 0.80
CheckLayoutsMethod · 0.80
RunMethod · 0.80
RunMethod · 0.80

Calls 2

IsFusionComputationMethod · 0.80
push_backMethod · 0.45

Tested by 3

CountMultiOutputFusionsFunction · 0.64
TEST_FFunction · 0.64