Searched refs:fused_instruction (Results 1 – 6 of 6) 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()
47 for (auto* fused_instruction : in instructions_to_modify() local50 << fused_instruction->ToString(); in instructions_to_modify()51 if (instruction_filter_function_(fused_instruction)) { in instructions_to_modify()
1541 for (auto* fused_instruction : in SetFusionLayouts() local1543 if (fused_instruction->opcode() == HloOpcode::kParameter) { in SetFusionLayouts()1545 fusion->operand(fused_instruction->parameter_number()); in SetFusionLayouts()1547 fused_instruction->shape())); in SetFusionLayouts()1549 fusion_operand->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()1550 } else if (fused_instruction == fusion->fused_expression_root()) { in SetFusionLayouts()1554 ShapeUtil::Compatible(fusion->shape(), fused_instruction->shape())); in SetFusionLayouts()1556 fusion->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()1557 } else if (fused_instruction->opcode() == HloOpcode::kGetTupleElement) { in SetFusionLayouts()1561 fused_instruction->operand(0)->shape().tuple_shapes( in SetFusionLayouts()[all …]
215 for (const auto& fused_instruction : instruction->fused_instructions()) { in PerformSinkReshapeOrTranspose() local216 Shape* shape = fused_instruction->mutable_shape(); in PerformSinkReshapeOrTranspose()
1247 auto fused_instruction = *fused_it; in MergeFusionInstruction() local1248 if (fused_instruction->opcode() == HloOpcode::kParameter) { in MergeFusionInstruction()1250 fused_instruction->ReplaceAllUsesWith(cloned_fusion->mutable_operand( in MergeFusionInstruction()1251 fused_instruction->parameter_number()))); in MergeFusionInstruction()1253 unfused_instructions.push_back(fused_instruction); in MergeFusionInstruction()1283 auto fused_instruction = *rit; in MergeFusionInstructionIntoMultiOutput() local1284 if (fused_instruction->opcode() == HloOpcode::kParameter) { in MergeFusionInstructionIntoMultiOutput()1285 InsertOrDie(&old_to_new, fused_instruction, in MergeFusionInstructionIntoMultiOutput()1287 fused_instruction->parameter_number())); in MergeFusionInstructionIntoMultiOutput()1294 parent()->AddInstruction(fused_instruction->Clone()); in MergeFusionInstructionIntoMultiOutput()[all …]
90 for (auto* fused_instruction : fusion->fused_instructions()) { in CalculateBytesReadByFusionInstruction() local91 if (fused_instruction->opcode() != HloOpcode::kParameter) { in CalculateBytesReadByFusionInstruction()94 bytes += CalculateBytesReadByFusionParameter(fused_instruction); in CalculateBytesReadByFusionInstruction()