Searched defs:fusion_kind (Results 1 – 7 of 7) sorted by relevance
66 string fusion_kind = 11; field
789 FusionKind fusion_kind() const { return fusion_kind_; } in fusion_kind() function
1201 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()1207 const Shape& shape, FusionKind fusion_kind, in CreateFusion()3355 HloInstruction::FusionKind HloInstruction::fusion_kind() const { in fusion_kind() function in xla::HloInstruction
616 HloInstruction::FusionKind fusion_kind) { in CreateFusionInstruction()
1134 FusionKind fusion_kind, in HloFusionInstruction()1145 const Shape& shape, FusionKind fusion_kind, in HloFusionInstruction()
1378 optional<HloInstruction::FusionKind> fusion_kind; in ParseInstructionRhs() local
264 HloInstruction::FusionKind fusion_kind = in RunFusionAndCheckOpcodesWereFused()