Searched refs:GetOutputsOfFusion (Results 1 – 1 of 1) sorted by relevance
33 absl::InlinedVector<HloInstruction*, 2> GetOutputsOfFusion( in GetOutputsOfFusion() function56 auto outputs = GetOutputsOfFusion(instr); in GetUniqueOutputTypeOfFusion()128 auto outputs = GetOutputsOfFusion(instr); in IsFusionSupported()385 auto old_output = GetOutputsOfFusion(*fused_fusion_instrs[j])[i]; in CreateFusedComputation()407 auto old_output = GetOutputsOfFusion(*fused_fusion_instrs[j])[i]; in CreateFusedComputation()451 auto output = GetOutputsOfFusion(*fused_instr)[j]; in Fuse()