Searched defs:fused_root (Results 1 – 5 of 5) sorted by relevance
48 const HloInstruction* fused_root = instr->fused_expression_root(); in MayBeImplementedAsInPlaceDynamicUpdateSlice() local75 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace() local
1463 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local
1239 HloInstruction* fused_root) in HloFusionInstruction()1599 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local
1306 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
2922 const HloInstruction* fused_root = unnested_hlo->fused_expression_root(); in IsUnrollingColumnReductionBeneficial() local