Searched refs:set_fusion_kind (Results 1 – 5 of 5) sorted by relevance
542 fusion_instruction->set_fusion_kind(kind); in AddFusionInstruction()
1483 void set_fusion_kind(FusionKind kind);
791 void set_fusion_kind(FusionKind kind) { fusion_kind_ = kind; } in set_fusion_kind() function
3359 void HloInstruction::set_fusion_kind(FusionKind kind) { in set_fusion_kind() function in xla::HloInstruction3360 return Cast<HloFusionInstruction>(this)->set_fusion_kind(kind); in set_fusion_kind()
1172 proto.set_fusion_kind(xla::ToString(fusion_kind())); in ToProto()