Searched defs:fused_root (Results 1 – 5 of 5) sorted by relevance
46 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace() local
1195 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local
1135 HloInstruction* fused_root) in HloFusionInstruction()1475 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local
1201 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
3613 const HloInstruction* fused_root = unnested_hlo->fused_expression_root(); in IsUnrollingColumnReductionBeneficial() local