Searched refs:fused_instruction (Results 1 – 5 of 5) sorted by relevance
65 for (HloInstruction* fused_instruction : in Defuse()67 if (fused_instruction->opcode() == HloOpcode::kParameter) { in Defuse()71 for (HloInstruction* operand : fused_instruction->operands()) { in Defuse()76 fused_instruction->CloneWithNewOperands(fused_instruction->shape(), in Defuse()78 defused_instructions[fused_instruction] = defused_instruction; in Defuse()
1682 for (auto* fused_instruction : in SetFusionLayouts() local1684 if (fused_instruction->opcode() == HloOpcode::kParameter) { in SetFusionLayouts()1686 fusion->operand(fused_instruction->parameter_number()); in SetFusionLayouts()1688 fused_instruction->shape())); in SetFusionLayouts()1690 fusion_operand->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()1691 } else if (fused_instruction == fusion->fused_expression_root()) { in SetFusionLayouts()1695 ShapeUtil::Compatible(fusion->shape(), fused_instruction->shape())); in SetFusionLayouts()1697 fusion->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()1698 } else if (fused_instruction->opcode() == HloOpcode::kGetTupleElement) { in SetFusionLayouts()1702 fused_instruction->operand(0)->shape().tuple_shapes( in SetFusionLayouts()[all …]
216 for (const auto& fused_instruction : instruction->fused_instructions()) { in PerformSinkReshapeOrTranspose() local217 Shape* shape = fused_instruction->mutable_shape(); in PerformSinkReshapeOrTranspose()
1477 auto fused_instruction = *fused_it; in MergeFusionInstruction() local1478 if (fused_instruction->opcode() == HloOpcode::kParameter) { in MergeFusionInstruction()1480 fused_instruction->ReplaceAllUsesWith(cloned_fusion->mutable_operand( in MergeFusionInstruction()1481 fused_instruction->parameter_number()))); in MergeFusionInstruction()1483 unfused_instructions.push_back(fused_instruction); in MergeFusionInstruction()1540 auto fused_instruction = *rit; in MergeFusionInstructionIntoMultiOutput() local1541 if (fused_instruction->opcode() == HloOpcode::kParameter) { in MergeFusionInstructionIntoMultiOutput()1542 InsertOrDie(&old_to_new, fused_instruction, in MergeFusionInstructionIntoMultiOutput()1544 fused_instruction->parameter_number())); in MergeFusionInstructionIntoMultiOutput()1551 parent()->AddInstruction(fused_instruction->Clone()); in MergeFusionInstructionIntoMultiOutput()[all …]
91 for (auto* fused_instruction : fusion->fused_instructions()) { in CalculateBytesReadByFusionInstruction() local92 if (fused_instruction->opcode() != HloOpcode::kParameter) { in CalculateBytesReadByFusionInstruction()95 bytes += CalculateBytesReadByFusionParameter(fused_instruction); in CalculateBytesReadByFusionInstruction()