Searched refs:consumer_for_fusion (Results 1 – 1 of 1) sorted by relevance
284 auto* consumer_for_fusion = SelectPreferredFusionCandidate(candidates); in DoMultiOutputFusion() local285 if (consumer_for_fusion == nullptr) { in DoMultiOutputFusion()290 consumer_for_fusion->name()); in DoMultiOutputFusion()295 if (consumer_for_fusion->opcode() == HloOpcode::kFusion) { in DoMultiOutputFusion()297 << consumer_for_fusion->name(); in DoMultiOutputFusion()299 consumer_for_fusion->MergeFusionInstructionIntoMultiOutput(producer); in DoMultiOutputFusion()301 consumer_for_fusion->FuseInstructionIntoMultiOutput(producer); in DoMultiOutputFusion()312 consumer_for_fusion->shape(), in DoMultiOutputFusion()313 ChooseFusionKind(*producer, *consumer_for_fusion), in DoMultiOutputFusion()314 consumer_for_fusion)); in DoMultiOutputFusion()[all …]