Searched refs:HloFusionInstruction (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 1366 HloFusionInstruction::HloFusionInstruction(const Shape& shape, in HloFusionInstruction() function in xla::HloFusionInstruction
1377 HloFusionInstruction::HloFusionInstruction( in HloFusionInstruction() function in xla::HloFusionInstruction
1390 string HloFusionInstruction::ToCategory() const { in ToCategory()
1403 HloInstructionProto HloFusionInstruction::ToProto() const { in ToProto()
1411 bool HloFusionInstruction::IsElementwiseImpl( in IsElementwiseImpl()
1447 HloInstruction* HloFusionInstruction::AddFusionOperand( in AddFusionOperand()
1461 void HloFusionInstruction::MergeFusionInstruction( in MergeFusionInstruction()
1462 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstruction()
1466 HloFusionInstruction* cloned_fusion = in MergeFusionInstruction()
1467 static_cast<HloFusionInstruction*>(cloned.get()); in MergeFusionInstruction()
[all …]
|
| D | hlo_instruction.cc | 1486 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, in CreateFusion()
1494 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, operands, in CreateFusion()
2324 Cast<HloFusionInstruction>(user)->DeduplicateFusionOperands()); in ReplaceUseWithDifferentShape()
2404 Cast<HloFusionInstruction>(user)->DeduplicateFusionOperands()); in ReplaceAllUsesWithDifferentShape()
3984 return Cast<HloFusionInstruction>(this)->AddFusionOperand(new_operand); in AddFusionOperand()
3990 return Cast<HloFusionInstruction>(this)->MergeFusionInstruction( in MergeFusionInstruction()
3991 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstruction()
3997 return Cast<HloFusionInstruction>(this) in MergeFusionInstructionIntoMultiOutput()
3999 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()
4004 return Cast<HloFusionInstruction>(this)->FuseInstruction(instruction_to_fuse); in FuseInstruction()
[all …]
|
| D | hlo_instructions.h | 889 class HloFusionInstruction : public HloInstruction {
891 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
894 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
909 void MergeFusionInstruction(HloFusionInstruction* instruction_to_merge);
917 HloFusionInstruction* instruction_to_merge);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 478 auto* fusion_instr = xla::Cast<xla::HloFusionInstruction>(instr); in EmitFusionOp()
|