Searched refs:fused_root (Results 1 – 6 of 6) sorted by relevance
48 const HloInstruction* fused_root = instr->fused_expression_root(); in MayBeImplementedAsInPlaceDynamicUpdateSlice() local49 return fused_root->opcode() == HloOpcode::kDynamicUpdateSlice && in MayBeImplementedAsInPlaceDynamicUpdateSlice()50 fused_root->operand(0)->LatestNonGteAncestor()->opcode() == in MayBeImplementedAsInPlaceDynamicUpdateSlice()75 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace() local79 fused_root->mutable_operand(0)->LatestNonGteAncestorAndIndex(); in CanEmitFusedDynamicUpdateSliceInPlace()
1368 HloInstruction* fused_root) in HloFusionInstruction() argument1370 CHECK(fused_root != nullptr); in HloFusionInstruction()1372 set_parent(fused_root->parent()); in HloFusionInstruction()1373 set_metadata(fused_root->metadata()); in HloFusionInstruction()1374 CloneAndFuseInternal(fused_root); in HloFusionInstruction()1744 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local1747 if (fused_root->opcode() == HloOpcode::kTuple) { in CloneAndFuseInternal()1748 tuple_elements = fused_root->operands(); in CloneAndFuseInternal()1750 tuple_elements.push_back(fused_root); in CloneAndFuseInternal()1764 if (fused_root->opcode() == HloOpcode::kTuple) { in CloneAndFuseInternal()[all …]
1632 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local1635 if (fused_root == instruction) { in CheckFusionInstruction()1665 if (fused_root->user_count() != 0) { in CheckFusionInstruction()1673 if (instruction != fused_root) { in CheckFusionInstruction()
968 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
892 HloInstruction* fused_root);
1485 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion() argument1487 fused_root); in CreateFusion()