Searched refs:input_fusion (Results 1 – 2 of 2) sorted by relevance
162 HloInstruction* input_fusion = in CreateFusion() local168 if (candidates_index_.contains(input_fusion)) { in CreateFusion()169 index = candidates_index_[input_fusion]; in CreateFusion()172 InsertOrDie(&candidates_index_, input_fusion, index); in CreateFusion()173 candidates_.emplace_back(input_fusion); in CreateFusion()174 all_fusion_candidates_.push_back(input_fusion); in CreateFusion()177 reachability_->Replace(base, input_fusion); in CreateFusion()178 TF_CHECK_OK(computation()->ReplaceInstruction(base, input_fusion)); in CreateFusion()179 return input_fusion; in CreateFusion()
310 HloInstruction* input_fusion = in DoMultiOutputFusion() local316 << consumer_for_fusion->name() << " into " << input_fusion->name(); in DoMultiOutputFusion()318 computation_->ReplaceInstruction(consumer_for_fusion, input_fusion)); in DoMultiOutputFusion()320 input_fusion->MergeFusionInstructionIntoMultiOutput(producer); in DoMultiOutputFusion()322 input_fusion->FuseInstructionIntoMultiOutput(producer); in DoMultiOutputFusion()