Searched refs:fused_root (Results 1 – 7 of 7) sorted by relevance
46 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace() local47 if (fused_root->opcode() != HloOpcode::kDynamicUpdateSlice || in CanEmitFusedDynamicUpdateSliceInPlace()56 fused_root->mutable_operand(0)->LatestNonGteAncestorAndIndex(); in CanEmitFusedDynamicUpdateSliceInPlace()
1135 HloInstruction* fused_root) in HloFusionInstruction() argument1137 CHECK(fused_root != nullptr); in HloFusionInstruction()1139 set_parent(fused_root->parent()); in HloFusionInstruction()1140 set_metadata(fused_root->metadata()); in HloFusionInstruction()1141 CloneAndFuseInternal(fused_root); in HloFusionInstruction()1475 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local1478 if (fused_root->opcode() == HloOpcode::kTuple) { in CloneAndFuseInternal()1479 tuple_elements = fused_root->operands(); in CloneAndFuseInternal()1481 tuple_elements.push_back(fused_root); in CloneAndFuseInternal()1495 if (fused_root->opcode() == HloOpcode::kTuple) { in CloneAndFuseInternal()[all …]
1195 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local1198 if (fused_root == instruction) { in CheckFusionInstruction()1228 if (fused_root->user_count() != 0) { in CheckFusionInstruction()1236 if (instruction != fused_root) { in CheckFusionInstruction()
749 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
708 HloInstruction* fused_root);
1201 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion() argument1203 fused_root); in CreateFusion()
3613 const HloInstruction* fused_root = unnested_hlo->fused_expression_root(); in IsUnrollingColumnReductionBeneficial() local3615 if (fused_root->opcode() == HloOpcode::kReduce) { in IsUnrollingColumnReductionBeneficial()3616 use_chain_endings.insert(fused_root); in IsUnrollingColumnReductionBeneficial()3620 CHECK_EQ(fused_root->opcode(), HloOpcode::kTuple); in IsUnrollingColumnReductionBeneficial()3621 for (const HloInstruction* instr : fused_root->operands()) { in IsUnrollingColumnReductionBeneficial()