Searched defs:fused_root (Results 1 – 4 of 4) 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
1632 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local
1368 HloInstruction* fused_root) in HloFusionInstruction()1744 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local
1485 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()