Searched refs:MultiOutputFusion (Results 1 – 4 of 4) sorted by relevance
31 StatusOr<bool> MultiOutputFusion::Run(HloModule* module) { in Run()138 HloInstruction* MultiOutputFusion::Fuse(HloInstruction* instr1, in Fuse()160 HloInstruction* MultiOutputFusion::CreateFusion(HloInstruction* base, in CreateFusion()182 bool MultiOutputFusion::IsProfitableOperand(HloInstruction* instr) { in IsProfitableOperand()199 MultiOutputFusion::GetNewFusibles(HloInstruction* instr1, in GetNewFusibles()246 void MultiOutputFusion::UpdateBeforeFuse(HloInstruction* instr1, in UpdateBeforeFuse()269 void MultiOutputFusion::UpdateAfterFuse( in UpdateAfterFuse()298 bool MultiOutputFusion::LegalToFuse(HloInstruction* instr1, in LegalToFuse()306 bool MultiOutputFusion::LegalToFuseMainConstraints(HloInstruction* instr1, in LegalToFuseMainConstraints()360 void MultiOutputFusion::RecomputeReachability() { in RecomputeReachability()[all …]
49 class MultiOutputFusion : public HloModulePass {51 MultiOutputFusion() = default;
3212 TEST_P(MemorySpaceAssignmentTest, MultiOutputFusion) { in TEST_P() argument
420 TEST_F(InstructionFusionTest, MultiOutputFusion) { in TEST_F() argument