Searched refs:HloFusionInstruction (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 1133 HloFusionInstruction::HloFusionInstruction(const Shape& shape, in HloFusionInstruction() function in xla::HloFusionInstruction
1144 HloFusionInstruction::HloFusionInstruction( in HloFusionInstruction() function in xla::HloFusionInstruction
1157 string HloFusionInstruction::ToCategory() const { in ToCategory()
1170 HloInstructionProto HloFusionInstruction::ToProto() const { in ToProto()
1178 bool HloFusionInstruction::IsElementwiseImpl( in IsElementwiseImpl()
1214 HloInstruction* HloFusionInstruction::AddFusionOperand( in AddFusionOperand()
1231 void HloFusionInstruction::MergeFusionInstruction( in MergeFusionInstruction()
1232 HloFusionInstruction* instruction_to_merge) { in MergeFusionInstruction()
1236 HloFusionInstruction* cloned_fusion = in MergeFusionInstruction()
1237 static_cast<HloFusionInstruction*>(cloned.get()); in MergeFusionInstruction()
[all …]
|
| D | hlo_instruction.cc | 1202 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, in CreateFusion()
1210 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, operands, in CreateFusion()
1912 Cast<HloFusionInstruction>(user)->DeduplicateFusionOperands()); in ReplaceUseWithDifferentShape()
1971 Cast<HloFusionInstruction>(user)->DeduplicateFusionOperands()); in ReplaceAllUsesWithDifferentShape()
3289 return Cast<HloFusionInstruction>(this)->AddFusionOperand(new_operand); in AddFusionOperand()
3295 return Cast<HloFusionInstruction>(this)->MergeFusionInstruction( in MergeFusionInstruction()
3296 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstruction()
3302 return Cast<HloFusionInstruction>(this) in MergeFusionInstructionIntoMultiOutput()
3304 Cast<HloFusionInstruction>(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()
3309 return Cast<HloFusionInstruction>(this)->FuseInstruction(instruction_to_fuse); in FuseInstruction()
[all …]
|
| D | hlo_instructions.h | 705 class HloFusionInstruction : public HloInstruction {
707 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
710 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
725 void MergeFusionInstruction(HloFusionInstruction* instruction_to_merge);
733 HloFusionInstruction* instruction_to_merge);
|