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
2204 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local
1784 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
1813 HloInstruction* fused_root) in HloFusionInstruction()