Searched defs:fusion_kind (Results 1 – 9 of 9) sorted by relevance
127 string fusion_kind = 11; field
740 HloInstruction::FusionKind fusion_kind) { in CreateFusionInstruction()
1784 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()1789 const Shape& shape, FusionKind fusion_kind, in CreateFusion()4623 HloInstruction::FusionKind HloInstruction::fusion_kind() const { in fusion_kind() function in xla::HloInstruction
1302 FusionKind fusion_kind() const { return fusion_kind_; } in fusion_kind() function
1812 FusionKind fusion_kind, in HloFusionInstruction()1825 const Shape& shape, FusionKind fusion_kind, in HloFusionInstruction()
2425 optional<HloInstruction::FusionKind> fusion_kind; in CreateInstruction() local
248 HloInstruction::FusionKind fusion_kind = in RunFusionAndCheckOpcodesWereFused()
1495 auto fusion_kind = mlir::mhlo::symbolizeFusionKind( in ImportInstructionImpl() local
114 absl::string_view fusion_kind, in BuildFusion()