Searched refs:fusion_kind_ (Results 1 – 2 of 2) sorted by relevance
789 FusionKind fusion_kind() const { return fusion_kind_; } in fusion_kind()791 void set_fusion_kind(FusionKind kind) { fusion_kind_ = kind; } in set_fusion_kind()831 FusionKind fusion_kind_; variable
1136 : HloInstruction(HloOpcode::kFusion, shape), fusion_kind_(fusion_kind) { in HloFusionInstruction()1148 : HloInstruction(HloOpcode::kFusion, shape), fusion_kind_(fusion_kind) { in HloFusionInstruction()