Ë
    �…jì  ã                   ó>   — d dl mZ d dlmZ d dlmZ  G d„ de«      Zy)é    )ÚFusion)Úhelper)Ú	OnnxModelc                   ó*   ‡ — e Zd Zdefˆ fd„Zd„ Zˆ xZS )ÚFusionGeluApproximationÚmodelc                 ó.   •— t         ‰| �  |dddgd«       y )NÚFastGeluÚGeluÚBiasGeluÚGeluApproximation)ÚsuperÚ__init__)Úselfr   Ú	__class__s     €ús/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/onnxruntime/transformers/fusion_gelu_approximation.pyr   z FusionGeluApproximation.__init__   s   ø€ Ü‰Ñ˜ 
¨V°ZÐ,@ÐBUÕVó    c           
      ól  — t        j                  d|j                  |j                  | j                  j                  d|j                  dz   «      ¬«      }d|_        | j                  j                  |«       | j                  j                  |«       | j                  | j                  |j                  <   y )Nr
   Ú_Approximation)ÚinputsÚoutputsÚnamezcom.microsoft)r   Ú	make_nodeÚinputÚoutputr   Úcreate_node_nameÚop_typeÚdomainÚnodes_to_removeÚappendÚnodes_to_addÚthis_graph_nameÚnode_name_to_graph_namer   )r   ÚnodeÚinput_name_to_nodesÚoutput_name_to_nodeÚnew_nodes        r   ÚfusezFusionGeluApproximation.fuse   sŒ   € Ü×#Ñ#ØØ—:‘:Ø—K‘KØ—‘×,Ñ,¨Z¸¿¹ÐHXÑ9XÓYô	
ˆð *ˆŒØ×Ñ×#Ñ# DÔ)Ø×Ñ× Ñ  Ô*Ø6:×6JÑ6Jˆ×$Ñ$ X§]¡]Ò3r   )Ú__name__Ú
__module__Ú__qualname__r   r   r(   Ú__classcell__)r   s   @r   r   r      s   ø„ ðW˜iõ Wö
Kr   r   N)Úfusion_baser   Úonnxr   Ú
onnx_modelr   r   © r   r   ú<module>r1      s   ðõ Ý Ý  ôK˜fõ Kr   