Ng nddlZddlmZddlmZddlmZejeZ GddeZ dS)N)Fusion)helper) OnnxModelc*eZdZdeffd ZdZxZS)FusionQuickGelumodelcPt|ddgdS)N QuickGeluMul)super__init__)selfr __class__s e/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnxruntime/transformers/fusion_quickgelu.pyr zFusionQuickGelu.__init__s'  eW55555cR|jdkrtddS|}|jd}|j|dgdg}|tddS|d}|j|dgdg}|tddS|d}|j|jd}t|dz d krtd dS|jd|krtd dStj d |g|j dg|j d  } d| _ | jtjd|g|j|||g|j| |j|j| j<|d dS)Nr z/fuse_quickgelu: failed to match second Mul noderSigmoidz,fuse_quickgelu: failed to match Sigmoid nodez.fuse_quickgelu: failed to match first Mul nodegrBs   8 $ $;+;+;+;+;+f;+;+;+;+;+r