Ë
    [^(h&¡  ã            	       ó:	  — d dl Z d dlZd dlmc mZ d dlmZmZm	Z	m
Z
mZmZmZ d dlmZmZ  e j&                  ej(                  d¬«      Z ed«       e
j,                  ddd	«      dJd
ej.                  fd„«       «       Z ed«       e
j,                  ddd	«      dJd
ej.                  fd„«       «       Z ed«       e
j,                  ddd«      dKd
ej.                  fd„«       «       Z ed«       e
j,                  dddd«      dJd
ej.                  fd„«       «       Z ed«      dJd
ej.                  fd„«       Z ed«      	 dJd
ej.                  fd„«       Z ed«      	 dJd
ej.                  fd„«       Z ed«       e
j,                  dddd«      	 dJd
ej.                  fd„«       «       Z ed«       e
j,                  ddd«      dLd
ej.                  fd„«       «       Z  ed«      dJd
ej.                  fd„«       Z! ed«       e
j,                  dddd«      dMd
ej.                  fd„«       «       Z" ed «       e
j,                  dddddd«      	 	 dNd
ej.                  fd!„«       «       Z# ed"«       e
j,                  ddddd«      	 	 dNd
ej.                  fd#„«       «       Z$d$„ Z% ed% e
jL                  d&d'«      g¬(«      d)„ «       Z' ed*«      d
ej.                  fd+„«       Z( ed,«       e
j,                  dddd«      dJd
ej.                  fd-„«       «       Z) ed.«      d
ej.                  fd/„«       Z* ed0«      	 dOd
ej.                  fd1„«       Z+ ed2«       e
j,                  dddd«      d
ej.                  fd3„«       «       Z, ed4«      d
ej.                  fd5„«       Z- ed6«      d
ej.                  fd7„«       Z. ed8«      d
ej.                  fd9„«       Z/ ed:«      d
ej.                  fd;„«       Z0 ed<«      d
ej.                  fd=„«       Z1 ed>«      d
ej.                  fd?„«       Z2 ed@«      d
ej.                  fdA„«       Z3 edB«      d
ej.                  fdC„«       Z4 edD«      d
ej.                  fdE„«       Z5 edF«      d
ej.                  fdG„«       Z6 edH«      d
ej.                  fdI„«       Z7y)Pé    N)Ú
_constantsÚ_type_utilsÚerrorsÚsymbolic_helperÚsymbolic_opset11Úsymbolic_opset9Úutils)Ú	jit_utilsÚregistrationé   )Úopsetzaten::softmaxÚvÚiÚnoneÚgc                 ó  — | j                  d||¬«      }|rm|j                  «       j                  «       dk7  rLt        j                  |dd«      }| j                  d|t        j                  |«      j                  «       ¬«      }|S )NÚSoftmax©Úaxis_iúprim::Constantr   ÚdtypeÚCast©Úto_i©ÚopÚnodeÚkindr   Ú
_get_constr   ÚJitScalarTypeÚ	onnx_type)r   ÚinputÚdimr   ÚsoftmaxÚparsed_dtypes         úY/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/onnx/symbolic_opset13.pyr$   r$      s{   € ð �d‰d�9˜e¨CˆdÓ0€GÙ�—‘“×"Ñ"Ó$Ð(8Ò8Ü&×1Ñ1°%¸¸gÓFˆØ—$‘$Ø�G¤+×";Ñ";¸LÓ"I×"SÑ"SÓ"Uð ó 
ˆð €Nó    zaten::log_softmaxc                 ó  — | j                  d||¬«      }|rm|j                  «       j                  «       dk7  rLt        j                  |dd«      }| j                  d|t        j                  |«      j                  «       ¬«      }|S )NÚ
LogSoftmaxr   r   r   r   r   r   r   )r   r"   r#   r   Ú	return_opr%   s         r&   Úlog_softmaxr+   &   s|   € ð —‘�\ 5°�Ó5€IÙ�—‘“×"Ñ"Ó$Ð(8Ò8Ü&×1Ñ1°%¸¸gÓFˆØ—D‘DØ�I¤K×$=Ñ$=¸lÓ$K×$UÑ$UÓ$Wð ó 
ˆ	ð Ðr'   zaten::frobenius_normc                 ó  — t        j                  |d«      }t        j                  |«      s"t        |«      dk(  r| j	                  d|d¬«      S | j	                  d||«      }t        j
                  | |||¬«      }| j	                  d|«      S )NÚisr   ÚReduceL2©Ú
keepdims_iÚMulÚSqrt)r   Ú_maybe_get_constÚ	_is_valueÚlenr   Ú_reducesum_helper)r   Úselfr#   ÚkeepdimÚdim_valÚsqrÚsumsqrs          r&   Úfrobenius_normr<   2   s}   € ô ×.Ñ.¨s°DÓ9€GÜ×$Ñ$ WÔ-´#°g³,À!Ò2CØ�t‰t�J °ˆtÓ3Ð3Ø
�$‰$ˆu�d˜DÓ
!€CÜ×.Ñ.¨q°#°sÀwÔO€FØ�4‰4�˜ÓÐr'   zaten::splitc                 ó|  — t        j                  ||«      �s¥| j                  d|||¬«      }|€|S t        j                  |«      �rt	        t        j
                  |«      «      |k(  röt        j
                  |«      D �cg c]  }t        j                  | |dg«      ‘Œ }}| j                  dt        j                  dgt        j                  ¬«      ¬«      }| j                  dt        j                  |gt        j                  ¬«      ¬«      }	g }
t        |«      D ]>  }| j                  d|||   «      }|
j                  | j                  d||||	«      «       |}Œ@ |
S t        |«      D �cg c]J  }| j                  d	|| j                  dt        j                  |gt        j                  ¬«      ¬«      «      ‘ŒL c}S t        j                  |j                  «       d
«      }|j                  «       dkD  r| j                  d||||¬«      S t        j                  |dd«      }t        j                   ||«      }|€|�||z  }nt#        j$                  d|«      ‚|g||z  z  }||z  }|r|j                  |«       | j                  dt        j                  |«      ¬«      }| j                  d||||¬«      S c c}w c c}w )NÚSplitToSequencer   r   ÚConstant©r   ©Úvalue_tÚAddÚSliceÚ
SequenceAtÚvalueÚSplit©r   Úoutputsr   Ú
split_sizeú$Unknown dimension size not supported)r   Ú_is_split_staticr   Ú_is_packed_listr5   Ú_unpack_listÚ_unsqueeze_helperÚtorchÚtensorÚlongÚrangeÚappendÚ	_node_getr   r#   r   Ú_get_tensor_dim_sizer   ÚSymbolicValueError)r   r7   Úsplit_size_or_sizesr#   Ú_outputsÚ	split_outr   Úsplit_sizesÚstartÚaxisÚresr   ÚendÚ	split_valrJ   ÚsizeÚsplitsÚleftovers                     r&   Úsplitrd   =   s›  € ô ×+Ñ+Ð,?ÀÕJØ—D‘DÐ*¨DÐ2EÈc�DÓRˆ	ØÐØÐô ×+Ñ+Ð,?Õ@Ü”O×0Ñ0Ð1DÓEÓFÈ(ÒRô )×5Ñ5Ð6IÓJöàô  ×1Ñ1°!°Q¸¸Õ<ðˆKð ð
 —D‘D˜¬U¯\©\¸1¸#ÄUÇZÁZÔ-P�DÓQˆEØ—4‘4˜
¬E¯L©L¸#¸ÄeÇjÁjÔ,Q�4ÓRˆDØˆCÜ˜8“_ò �Ø—d‘dØ˜5 +¨a¡.ó�ð —
‘
˜1Ÿ4™4 ¨¨u°c¸4Ó@ÔAØ‘ðð ˆJô ˜8“_ö
ð ð �D‰DØØØ—‘�Z¬¯©°q°cÄÇÁÔ)L�ÓMõò
ð 	
ô  ×)Ñ)Ð*=×*BÑ*BÓ*DÀgÓN€IØ‡}�}ƒ˜ÒØ�t‰t�G˜TÐ#6¸sÈHˆtÓUÐUÜ ×+Ñ+Ð,?ÀÀlÓS€Jä×/Ñ/°°cÓ:€DØ€|ØÐØ Ñ(‰Dä×+Ñ+Ø6¸óð ð ˆ\˜T ZÑ/Ñ0€FØ�jÑ €HÙØ�‰�hÔØ�T‰T�*¤e§l¡l°6Ó&:ˆTÓ;€FØ�4‰4�˜˜v¨c¸8ˆ4ÓDÐDùòUùò
s   Á?J4Å+AJ9zaten::split_with_sizesc                 ó    — t        | ||||«      S ©N©rd   ©r   r7   r[   r#   rY   s        r&   Úsplit_with_sizesri   v   s   € ä��D˜+ s¨HÓ5Ð5r'   zaten::unsafe_splitc                 ó    — t        | ||||«      S rf   rg   )r   r7   rX   r#   rY   s        r&   Úunsafe_splitrk   {   s   € ô ��DÐ-¨s°HÓ=Ð=r'   zaten::unsafe_split_with_sizesc                 ó    — t        | ||||«      S rf   )ri   rh   s        r&   Úunsafe_split_with_sizesrm   ‚   s   € ô ˜A˜t [°#°xÓ@Ð@r'   zaten::tensor_splitc                 ót  — | j                  dt        j                  |t        j                  ¬«      ¬«      }t	        j
                  | |d«      }| j                  dt        j                  dt        j                  ¬«      ¬«      }t        j                  ||«      �rìt        j                  |j                  «       d«      }|j                  «       dkD  rý| j                  dt        j                  dgt        j                  ¬«      ¬«      }g }	|€J ‚t        |dz
  «      D ]r  }
| j                  d|| j                  dt        j                  |
gt        j                  ¬«      ¬«      d¬«      }|	j                  | j                  d	||||«      «       |}Œt t        j                  | ||«      }|	j                  | j                  d	||||«      «       |	S t        j                  |d
d«      }t        j                  ||«      }|€|�||z  }nt!        j"                  d|«      ‚||z  }||z  }||dz   gz  }||z
  |gz  }| j                  dt        j                  ||z   t        j                  ¬«      ¬«      }| j                  d||||¬«      S t        j$                  |«      �r¿t        j&                  |«      dk(  �r¦t        j                  | || j                  dt        j                  d«      ¬«      «      }t	        j
                  | |d«      }| j                  d|t(        j*                  j,                  ¬«      }| j                  dt        j                  dgt        j                  ¬«      ¬«      }| j                  d||d¬«      }| j                  d«      }t/        j0                  | d|||dd¬«      \  }\  }}|j2                  }t5        j6                  |«      }t5        j6                  |«      }t5        j6                  |«      }|j                  d||d¬«      }|j                  d||j                  d||«      d¬«      }|j                  d	||||«      }|j                  d||«      }|j                  d|«      }t5        j8                  ||«       t5        j8                  ||«       |j                  «       j;                  «       }| j                  d|| j                  dt        j                  dt        j                  ¬«      ¬«      d¬«      }t	        j
                  | |d«      }t        j                  | ||«      }| j                  d	||||«      }| j                  d||«      S t        j                  | ||«      } | j                  d| |«      }| j                  d||«      }!| j                  d| |«      }| j                  d|!|«      }| j                  d|| j                  dt	        j
                  | |d«      |«      «      }| j                  d||d¬«      }|€| j                  d|||¬«      S | j                  d||||¬«      S )Nr?   r@   rA   r   é   rF   ÚGatherr   rD   r   Úindices_or_sectionsrK   rG   rH   r   r   ÚConcatÚSequenceEmptyÚLoop)rI   Ún_blocksrC   ÚSequenceInsertÚIdentityéÿÿÿÿÚDivÚModÚTileÚSubr>   )r   rP   rQ   rR   Úopset11Ú	unsqueezer   rL   rU   r   r#   rS   rT   Ú_size_helperr   rV   r   rW   Ú
_is_tensorÚ_get_tensor_rankÚ_C_onnxÚTensorProtoDataTypeÚBOOLr
   Úadd_op_with_blocksÚblockr	   Ú_add_input_to_blockÚ_add_output_to_blockÚoutput)"r   r7   rq   r#   rY   r]   Úconst_1r`   r\   r^   r   r_   rJ   ra   Úmin_split_sizeÚnum_splits_one_extrarb   rc   Úloop_lenÚloop_conditionÚ	padding_0Úfinal_splitsÚloopÚloop_contextÚ_Ú
loop_blockÚblock_input_iterÚcondÚsliceÚcond_outÚloop_outÚ
last_sliceÚdim_sizeÚmin_split_size_plus_1s"                                     r&   Útensor_splitr�   ‰   sÑ  € ð
 �4‰4�
¤E§L¡L°¼E¿J¹JÔ$Gˆ4ÓH€DÜ×Ñ˜Q  aÓ(€DØ�d‰d�:¤u§|¡|°A¼U¿Z¹ZÔ'HˆdÓI€Gä×'Ñ'Ð(;¸XÕFÜ#×-Ñ-Ð.A×.FÑ.FÓ.HÈ'ÓRˆ	à�=‰=‹?˜QÒØ—D‘D˜¬U¯\©\¸1¸#ÄUÇZÁZÔ-P�DÓQˆEØˆCØÐ'Ð'Ð'Ü˜8 a™<Ó(ò �Ø—d‘dØØ'Ø—D‘D˜¬U¯\©\¸1¸#ÄUÇZÁZÔ-P�DÓQØð	 ó �ð —
‘
˜1Ÿ4™4 ¨¨u°c¸4Ó@ÔAØ‘ðô "×.Ñ.¨q°$¸Ó=ˆCØ�J‰J�q—t‘t˜G T¨5°#°tÓ<Ô=ØˆJä$×/Ñ/Ø Ð&;ó
ˆ
ô ×3Ñ3°D¸#Ó>ˆØˆ<ØÐ#Ø! HÑ,‘ä×/Ñ/Ø:¸Dóð ð  Ñ+ˆØ# jÑ0Ðà%¨¸!Ñ);Ð(<Ñ<ˆØÐ!5Ñ5¸.Ð9IÑIˆà—‘Ø¤§¡¨V°hÑ->ÄeÇjÁjÔ Qð ó 
ˆð �t‰t�G˜T 6°#¸xˆtÓHÐHô 	×"Ñ"Ð#6Õ7Ü×,Ñ,Ð-@ÓAÀQÓFä"×/Ñ/ØÐ" A§D¡D¨¼U¿\¹\È!»_ DÓ$Mó
ˆô ×$Ñ$ Q¨°!Ó4ˆØŸ™˜f g´G×4OÑ4O×4TÑ4T˜ÓUˆð —D‘D˜¬U¯\©\¸1¸#ÄUÇZÁZÔ-P�DÓQˆ	ØŸd™d 8¨YÐ8KÐTU˜dÓVÐà—t‘t˜OÓ,ˆä#,×#?Ñ#?Øˆv�x °ÀqÐSTô$
Ñ ˆ‰oˆ|˜qð "×'Ñ'ˆ
Ü ×4Ñ4°ZÓ@ÐÜ×(Ñ(¨Ó4ˆÜ×0Ñ0°Ó<ˆà—‘ØÐ)Ð+;ÀAð  ó 
ˆð �o‰oØØØ�O‰O˜EÐ#3°WÓ=Øð	 ó 
ˆð —‘ ¨¨u°c¸4Ó@ˆØ#—‘Ð'7¸ÀuÓMˆð  —?‘? :¨~Ó>ˆÜ×"Ñ" :¨xÔ8Ü×"Ñ" :¨|Ô<à—9‘9“;×%Ñ%Ó'ˆØ—‘ØØØ�D‰D�¤U§\¡\°"¼E¿J¹JÔ%GˆDÓHØð	 ó 
ˆô ×!Ñ! ! U¨AÓ.ˆÜ×*Ñ*¨1¨d°DÓ9ˆà—T‘T˜' 4¨°°TÓ:ˆ
à�t‰tÐ$ h°
Ó;Ð;ô #×/Ñ/°°4¸Ó>ˆØŸ™˜e XÐ/BÓCˆØ !§¡ØØØó!
Ðð
  !Ÿt™t E¨8Ð5HÓIÐØ—‘�fÐ3Ð5IÓJˆØ—4‘4ØØØ�D‰DØÜ×!Ñ! !Ð%8¸!Ó<Ø$óó
ˆð —‘�h ¨¸�Ó;ˆØÐØ—4‘4Ð)¨4°À�4ÓDÐDØ�t‰t�G˜T 6°#¸xˆtÓHÐHr'   zaten::unbindc                 óÆ  — |€J| j                  d|| j                  dt        j                  dt        j                  ¬«      ¬«      |d¬«      S | j                  dt        j                  dg|z  «      ¬«      }| j                  d||||¬	«      }|dk(  r|gn|}|D �cg c]:  }| j                  d
|| j                  dt        j                  |g«      ¬«      «      ‘Œ< }}|S c c}w )Nr>   r?   ro   r@   rA   r   ©r   r0   rG   rH   ÚSqueeze)r   rP   rQ   rR   )r   r7   r#   rY   rb   rI   ÚoutÚsqueezed_outputss           r&   Úunbindr£     sç   € ð ÐØ�t‰tØØØ�D‰D�¤U§\¡\°!¼5¿:¹:Ô%FˆDÓGØØð ó 
ð 	
ð �T‰T�*¤e§l¡l°A°3¸±>Ó&BˆTÓC€FØ�d‰d�7˜D &°¸hˆdÓG€GØ# qš=ˆw‰i¨g€Gð öàð 	
�‰ˆY˜˜QŸT™T *´e·l±lÀCÀ5Ó6I˜TÓJÕKðÐð ð Ðùò	s   Â?Czaten::nonzero_numpyc                 óH   — t        | t        j                  | |«      d|¬«      S )Nro   )rY   )r£   Úopset9Únonzero)r   r"   rY   s      r&   Únonzero_numpyr§   )  s    € ô �!”V—^‘^ A uÓ-¨q¸8ÔDÐDr'   zaten::wherec           
      óT  — t        j                  |«      s,| j                  d|t        j                  j
                  ¬«      }|€Rt        j                  | |«      }t        j                  | || j                  dt        j                  d«      ¬«      |«      S | j                  d|||«      S )Nr   r   r?   ro   rA   ÚWhere)r   Ú_is_boolr   r‚   rƒ   r„   r¥   r¦   Ú_unbind_helperrP   rQ   )r   Ú	conditionr7   ÚotherrY   s        r&   Úwherer®   /  s�   € ô ×#Ñ# IÔ.Ø—D‘D˜ ´×1LÑ1L×1QÑ1Q�DÓRˆ	Ø€|Ü—N‘N 1 iÓ0ˆ	Ü×-Ñ-Øˆy˜!Ÿ$™$˜z´5·<±<À³?˜$ÓCÀXó
ð 	
ð �4‰4�˜ D¨%Ó0Ð0r'   z&aten::fake_quantize_per_channel_affinec                 ó   — ||fdvrt        j                  d|› d|› d�|«      ‚|dk(  r-| j                  d|t        j                  j
                  ¬«      }n,| j                  d|t        j                  j                  ¬«      }| j                  d||||¬	«      }||fd
k(  r[| j                  d|t        j                  | «      | j                  dt        j                  dt        j                  ¬«      ¬«      «      }| j                  d||||¬	«      S )N©)r   éÿ   ©i€ÿÿÿé   ©r   r³   úVFor (quant_min, quant_max), ONNX allows only (0, 127), (0, 255) and (-128, 127). Got (ú, ú)r   r   r   ÚQuantizeLinearr   r´   ÚClipr?   r³   r@   rA   ÚDequantizeLinear)r   rW   r   r‚   rƒ   ÚUINT8ÚINT8r¥   ÚunusedrP   rQ   Úuint8)r   ÚinputsÚscaleÚ
zero_pointr]   Ú	quant_minÚ	quant_maxÚ	quantizeds           r&   Ú fake_quantize_per_channel_affinerÅ   =  s	  € ð 	�9ÐÐ%FÑFÜ×'Ñ'ðØ�;˜b  ¨1ð.àó
ð 	
ð �A‚~Ø—T‘T˜& *´7×3NÑ3N×3TÑ3T�TÓU‰
à—T‘T˜& *´7×3NÑ3N×3SÑ3S�TÓTˆ
Ø—‘Ð% v¨u°jÈ�ÓN€IØ�9Ð Ò)Ø—D‘DØØÜ�M‰M˜!ÓØ�D‰D�¤U§\¡\°#¼U¿[¹[Ô%IˆDÓJó	
ˆ	ð �4‰4Ð" I¨u°jÈˆ4ÓNÐNr'   z%aten::fake_quantize_per_tensor_affinec                 ó  — ||fdvrt        j                  d|› d|› d�|«      ‚|dk(  r-| j                  d|t        j                  j
                  ¬«      }n,| j                  d|t        j                  j                  ¬«      }t        j                  j                  |t        j                  j                  «      t        j                  j                  k7  r,| j                  d|t        j                  j                  ¬«      }| j                  d|||«      }||fd	k(  r[| j                  d
|t        j                  | «      | j                  dt        j                  dt        j                   ¬«      ¬«      «      }| j                  d|||«      S )Nr°   rµ   r¶   r·   r   r   r   r¸   r´   r¹   r?   r³   r@   rA   rº   )r   rW   r   r‚   rƒ   r»   r¼   r   r    Ú
from_valueÚ	UNDEFINEDÚFLOATr¥   r½   rP   rQ   r¾   )r   r¿   rÀ   rÁ   rÂ   rÃ   rÄ   s          r&   Úfake_quantize_per_tensor_affinerÊ   `  sY  € ð 	�9ÐÐ%FÑFÜ×'Ñ'ðØ�;˜b  ¨1ð.àó
ð 	
ð
 �A‚~Ø—T‘T˜& *´7×3NÑ3N×3TÑ3T�TÓU‰
à—T‘T˜& *´7×3NÑ3N×3SÑ3S�TÓTˆ
ä×!Ñ!×,Ñ,¨U´K×4MÑ4M×4WÑ4WÓXÜ×$Ñ$×*Ñ*ò	+ð —‘�V˜U¬×)DÑ)D×)JÑ)J�ÓKˆØ—‘Ð% v¨u°jÓA€IØ�9Ð Ò)Ø—D‘DØØÜ�M‰M˜!ÓØ�D‰D�¤U§\¡\°#¼U¿[¹[Ô%IˆDÓJó	
ˆ	ð �4‰4Ð" I¨u°jÓAÐAr'   c                 ó   ‡ — dˆ fd„	}|S )Nc                 óº   •— t        j                  | |«      }|€t        j                  | |‰«      S t        j                  |dd«      }| j	                  ‰|||¬«      S )Nr   r8   r/   )r   Ú_maybe_cast_reduce_op_inputÚ_handle_reduce_dim_noner   r   )r   r7   r#   r8   Úonnx_op_names       €r&   Úsymbolicz%_reduce_op_symbolic.<locals>.symbolic‡  sZ   ø€ Ü×:Ñ:¸1¸dÓCˆØˆ;ä"×:Ñ:¸1¸dÀLÓQÐQä%×0Ñ0°¸#¸yÓIˆGØ—4‘4˜ d¨C¸G�4ÓDÐDr'   ©NN© )rÏ   rÐ   s   ` r&   Ú_reduce_op_symbolicrÓ   †  s   ø€ õEð €Or'   z	aten::sumÚ	ReduceSumÚsum)Údecoratec                 óR   ‡‡— t        | «      Št        j                  ˆˆfd„«       }|S )Nc                 óŒ   •— t        j                  dd«      ˆˆfd„«       }t        j                  dddd«      ˆˆfd„«       }||fS )Nr   r   c                 óþ  •— d }|j                  «       j                  «       dk(  rOt        j                  |dd«      }t	        j
                  |«      j                  «       }| j                  d||¬«      }n8|j                  «       j                  «       dk7  rt        j                  ‰d|«      S  ‰| |«      }|�Ft        j
                  j                  |«      j                  «       }||k7  r| j                  d||¬«      }|S ©Nzonnx::Constantr   r   r   r   r   ©
r   r   r   r   r   r    r!   r   Ú_unimplementedrÇ   )r   r7   r   Ú
dtype_onnxÚresultÚresult_dtype_onnxÚnamerÐ   s         €€r&   Úreduce_nodimz8_reduce_with_dtype.<locals>.reduce.<locals>.reduce_nodimœ  sæ   ø€ àˆJØ�z‰z‹|× Ñ Ó"Ð&6Ò6Ü'×2Ñ2°5¸#¸wÓG�Ü(×6Ñ6°uÓ=×GÑGÓI�
Ø—t‘t˜F D¨z�tÓ:‘Ø—‘“×"Ñ"Ó$Ð(8Ò8Ü&×5Ñ5°d¸GÀUÓKÐKÙ˜a Ó&ˆFØÐ%Ü$/×$=Ñ$=×$HÑ$HØó%ç‘)“+ð "ð %¨
Ò2ØŸT™T &¨&°z˜TÓB�FØˆMr'   r   c                 ó  •— d }|j                  «       j                  «       dk(  rOt        j                  |dd«      }t	        j
                  |«      j                  «       }| j                  d||¬«      }n8|j                  «       j                  «       dk7  rt        j                  ‰d|«      S  ‰	| |||«      }|�Ft        j
                  j                  |«      j                  «       }||k7  r| j                  d||¬«      }|S rÚ   rÛ   )
r   r7   r#   r8   r   rÝ   rÞ   rß   rà   rÐ   s
           €€r&   Ú
reduce_dimz6_reduce_with_dtype.<locals>.reduce.<locals>.reduce_dim®  sê   ø€ àˆJØ�z‰z‹|× Ñ Ó"Ð&6Ò6Ü'×2Ñ2°5¸#¸wÓG�Ü(×6Ñ6°uÓ=×GÑGÓI�
Ø—t‘t˜F D¨z�tÓ:‘Ø—‘“×"Ñ"Ó$Ð(8Ò8Ü&×5Ñ5°d¸GÀUÓKÐKÙ˜a  s¨GÓ4ˆFØÐ%Ü$/×$=Ñ$=×$HÑ$HØó%ç‘)“+ð "ð %¨
Ò2ØŸT™T &¨&°z˜TÓB�FØˆMr'   )r   Ú
parse_args)r   ÚargsÚkwargsrá   rã   rà   rÐ   s        €€r&   Úreducez"_reduce_with_dtype.<locals>.reduceš  sT   ø€ ä	×	#Ñ	# C¨Ó	0ô	ó 
1ð	ô" 
×	#Ñ	# C¨¨c°6Ó	:ô	ó 
;ð	ð" ˜ZÐ'Ð'r'   )rÓ   r   Ú_overload_by_arg_count)Úonnx_oprà   rç   rÐ   s    ` @r&   Ú_reduce_with_dtyperê   “  s/   ù€ ô
 # 7Ó+€Hä×+Ñ+ô%(ó ,ð%(ðN €Mr'   zaten::unflattenc                 óŒ  — t        j                  |«      }|€t        j                  dd«      S | j                  dt	        j
                  |gt        j                  ¬«      ¬«      }| j                  d||«      }| j                  d||«      }| j                  d|«      }| j                  dt	        j
                  d	gt        j                  ¬«      ¬«      }| j                  d
|| j                  dt	        j
                  dgt        j                  ¬«      ¬«      «      }| j                  d|||«      }| j                  d|| j                  dt	        j
                  dgt        j                  ¬«      ¬«      «      }	| j                  d
|	| j                  dt	        j
                  dgt        j                  ¬«      ¬«      «      }
| j                  dt	        j
                  t        j                  gt        j                  ¬«      ¬«      }| j                  d||
|«      }| j                  d|||d	¬«      }t        j                  | ||«      S )Nr#   zfONNX and PyTorch use different strategies to split the input. Input rank must be known at export time.r?   r@   rA   rC   rz   ÚShaper   ÚReshapero   rD   rr   r   )
r   r�   rÜ   r   rP   rQ   Úint64r   Ú	INT64_MAXÚ_reshape_helper)r   r"   r#   Úunflattened_sizeÚ	input_dimÚ
input_sizeÚhead_start_idxÚhead_end_idxÚhead_part_rankÚdim_plus_oneÚtail_start_idxÚtail_end_idxÚtail_part_rankÚfinal_shapes                 r&   Ú	unflattenrü   È  sà  € ä×0Ñ0°Ó7€IØÐÜ×-Ñ-Øð7ó
ð 	
ð —‘�Z¬¯©°y°kÌÏÉÔ)U�ÓV€IØ
�$‰$ˆu�i Ó
%€CØ
�$‰$ˆu�c˜9Ó
%€Cà—‘�g˜uÓ%€Jà—T‘T˜*¬e¯l©l¸A¸3ÄeÇkÁkÔ.R�TÓS€NØ—4‘4Ø�3˜Ÿ™˜Z´·±¸q¸cÌÏÉÔ1U˜ÓVó€Lð —T‘T˜' :¨~¸|ÓL€Nà—4‘4Øˆs�A—D‘D˜¬U¯\©\¸1¸#ÄUÇ[Á[Ô-Q�DÓRó€Lð —T‘TØØØ	�‰ˆZ¤§¡¨q¨c¼¿¹Ô!EˆÓFó€Nð
 —4‘4ØœEŸL™L¬*×*>Ñ*>Ð)?ÄuÇ{Á{ÔSð ó €Lð —T‘T˜' :¨~¸|ÓL€Nà—$‘$Ø�.Ð"2°NÈ1ð ó €Kô ×*Ñ*¨1¨e°[ÓAÐAr'   zaten::unsafe_chunkc                 óæ  — |€J| j                  d|| j                  dt        j                  dt        j                  ¬«      ¬«      |d¬«      S t	        j
                  ||«      }|€t	        j                  dd	«      S ||z   dz
  |z  }|g||z  z  }||z  }|r|j                  |«       | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d
||||¬«      S )Nr>   r?   ro   r@   rA   r   rŸ   Úunsafe_chunkzunknown dimension sizerG   rH   )r   rP   rQ   rR   r   rV   rÜ   rT   )	r   r7   Úchunksr#   rY   ra   rJ   rb   rc   s	            r&   rþ   rþ   ó  sñ   € ð ÐØ�t‰tØØØ�D‰D�¤U§\¡\°!¼5¿:¹:Ô%FˆDÓGØØð ó 
ð 	
ô ×/Ñ/°°cÓ:€DØ€|Ü×-Ñ-¨nÐ>VÓWÐWØ˜‘- !Ñ#¨Ñ.€JØˆ\˜T ZÑ/Ñ0€FØ�jÑ €HÙØ�‰�hÔð
 �T‰T�*¤e§l¡l°6ÄÇÁÔ&LˆTÓM€FØ�4‰4�˜˜v¨c¸8ˆ4ÓDÐDr'   z
aten::tilec                 ó¾  — | j                  d|«      }| j                  d|«      }| j                  d|«      }| j                  d||«      }| j                  dt        j                  dg«      ¬«      }| j                  d||«      }t        j                  | d|d	d
¬«      \  }	\  }
}}|
j                  dt        j
                  d
g«      ¬«      }|
j                  d||«      }|
j                  d||«      }|
j                  d||d¬«      }t        j                  |
j                  |«       |j                  d|«      }t        j                  |j                  |«       |	j                  «       j                  «       }| j                  d||«      }t        j                  | d|d	d
¬«      \  }\  }}}|j                  dt        j
                  d
g«      ¬«      }|j                  d|j                  d|«      |«      }|j                  d||«      }|j                  d||d¬«      }|j                  d||«      }t        j                  |j                  |«       |j                  d|«      }t        j                  |j                  |«       |j                  «       j                  «       }| j                  d|t        j                  j                  ¬«      }| j                  d||«      S )Nrì   ÚSizer|   r?   r   rA   ÚGreaterÚIfé   ro   )ru   rI   rí   ÚExpandrr   r   rw   ÚLessÚAbsr   r   r{   )r   rP   rQ   r
   r…   Ú
LongTensorr	   rˆ   r†   r   r‰   r‚   rƒ   ÚINT64)r   r7   ÚdimsÚ
self_shapeÚ	self_rankÚ	dims_rankÚdiffÚ
const_zeroÚdims_shorter_than_self_shapeÚif_op_greaterÚif_context_greaterÚelse_context_greaterr“   Ú	const_oneÚdiff_1d_greaterÚexapnd_ones_greaterÚdims_Úidentity_dimÚ
dims_finalÚdims_longer_than_self_shapeÚ
if_op_lessÚif_context_lessÚelse_context_lessÚdiff_1d_lessÚexapnd_ones_lessÚself_final_shapeÚself_Úidentity_selfÚ
self_finals                                r&   Útiler$    sÇ  € à—‘�g˜tÓ$€JØ—‘�V˜ZÓ(€IØ—‘�V˜TÓ"€IØ�4‰4��y )Ó,€DØ—‘�j¬%¯,©,¸°sÓ*;�Ó<€Jð $%§4¡4¨	°4¸Ó#DÐ ô
 	×$Ñ$Ø	ˆ4Ð-¸À1ô	ñ	ØÙ2Ð	Ð1Ø	ð #×%Ñ% j¼%×:JÑ:JÈAÈ3Ó:OÐ%ÓP€IØ(×+Ñ+¨I°t¸YÓG€OØ,×/Ñ/°¸)À_ÓUÐØ×!Ñ! (Ð,?ÀÈaÐ!ÓP€EÜ	×ÑÐ1×7Ñ7¸Ô?Ø'×*Ñ*¨:°tÓ<€LÜ	×ÑÐ3×9Ñ9¸<ÔHØ×#Ñ#Ó%×,Ñ,Ó.€Jð #$§$¡$ v¨t°ZÓ"@Ðô
 	×$Ñ$Ø	ˆ4Ð,°qÀ!ô	ñ	ØÙ,ˆÐ+Ø	ð  ×"Ñ" :´u×7GÑ7GÈÈÓ7LÐ"ÓM€IØ"×%Ñ%ØØ×Ñ˜5 $Ó'Øó€Lð
 '×)Ñ)¨(°I¸|ÓLÐØ&×)Ñ)ØÐ" J°qð *ó Ðð ×Ñ˜y¨$Ð0@ÓA€EÜ	×Ñ˜×4Ñ4°eÔ<Ø%×(Ñ(¨°TÓ:€MÜ	×ÑÐ0×6Ñ6¸ÔFØ—‘Ó"×)Ñ)Ó+€Jà—‘�f˜j¬w×/JÑ/J×/PÑ/P�ÓQ€JØ�4‰4�˜
 JÓ/Ð/r'   zaten::repeat_interleavec           
      ó²
  — t        j                  |«      }t        j                  |«      }t        j                  |«      }|€t        j                  d|«      ‚|€t        j                  d|«      ‚|€t        j                  d|«      ‚|}t        j
                  |«      rbt        j                  | || j                  dt        j                  dg«      ¬«      «      }t        j                  dt        j                  ¬«      }nt        j                  |«      }|dk  r|t        |«      z  }|j                  «       }	t        |«      D ]  \  }
}|�Œ	d	\  |	|
<   ||
<   Œ |dk(  s|d
k(  r |d   d
k(  rt        j                  | |||«      S |d
k(  xr |d   d u }|	|   dk(  s|rÄt        j                   | ||«      }t#        j$                  | |d«      }|r¬t        j                   | || j                  dt        j&                  dg«      ¬«      «      }| j                  d|| j                  dt        j&                  d
g«      ¬«      «      }t)        | || j                  d||«      |«      }nt+        j,                  | |||«      S | j                  d| j                  d|«      t        j                  d
gt        j.                  ¬«      ¬«      }t1        | ||d«      }t1        | |||«      }d\  |	|<   ||<   | j                  dt        j                  d
«      ¬«      }| j                  d|t2        j4                  j6                  ¬«      }|}| j                  d«      }t9        j:                  | d|||d
¬«      \  }\  }}|j<                  }t?        j@                  |«      }t?        j@                  |«      }t?        j@                  |«      }|j                  d||«      }|j                  d||«      }t#        j$                  |||d
z   «      }|j                  dt        j&                  |d |d
z    «      ¬«      ||j                  dt        j&                  ||d
z   d  «      ¬«      g} |j                  dg|¢­ddiŽ}t+        jB                  |||d «      }t        j                  ||| j                  dt        j&                  |	«      ¬«      «      }|j                  d||«      }|j                  d|t2        j4                  j6                  ¬«      }t?        jD                  ||«       t?        jD                  ||«       |jG                  «       jI                  «       } | j                  d| |¬«      } | S )NzGUnsupported: ONNX export of repeat_interleave for unknown repeats rank.zGUnsupported: ONNX export of repeat_interleave for unknown repeats size.zEUnsupported: ONNX export of repeat_interleave for unknown input size.r?   rx   rA   r   r@   )r   rx   ro   ÚEqualr  ÚConstantOfShaperì   )rx   ro   r   r   rs   rt   ©ru   rE   rr   r   rv   ÚConcatFromSequencer   )%r   r�   Ú_get_tensor_sizesr   rW   Ú_is_nonerð   r   rP   rQ   rî   Ú_maybe_get_scalarr5   ÚcopyÚ	enumerateÚ-_repeat_interleave_single_value_repeat_helperr   r}   r~   r  r®   r¥   Úrepeat_interleaverR   rd   r‚   rƒ   r„   r
   r…   r†   r	   r‡   Úexpandrˆ   r   r‰   )!r   r7   Úrepeatsr#   Úoutput_sizeÚrepeats_dimÚrepeats_sizesÚinput_sizesÚ	final_dimÚoutput_sizesÚidxró   Úcond_dynamic_repeatsÚrepsÚ
repeat_dimÚrepeat_condÚ	reps_likeÚr_splitsÚi_splitsrŽ   r�   r�   r‘   r’   r“   r”   r•   r–   Úr_splitÚi_splitÚr_concatr˜   r™   s!                                    r&   r0  r0  F  s,  € ô "×2Ñ2°7Ó;€KÜ#×5Ñ5°gÓ>€MÜ!×3Ñ3°DÓ9€KØÐÜ×'Ñ'ØUØó
ð 	
ð ÐÜ×'Ñ'ØUØó
ð 	
ð ÐÜ×'Ñ'ØSØó
ð 	
ð
 €Iô ×Ñ Ô$Ü×.Ñ.Øˆt�Q—T‘T˜*¬e¯l©l¸B¸4Ó.@�TÓAó
ˆô �l‰l˜1¤E§K¡KÔ0‰ä×/Ñ/°Ó4ˆð ˆQ‚wØŒs�;ÓÑˆà×#Ñ#Ó%€LÜ$ [Ó1ò 8‰ˆˆZØÑØ27Ñ/ˆL˜Ñ˜{¨3Ò/ð8ð
 �aÒ˜K¨1Ò,°¸qÑ1AÀQÒ1FÜ×LÑLØˆt�W˜có
ð 	
ð '¨!Ñ+ÒH°¸aÑ0@ÀDÐ0HÐà�CÑ˜AÒÑ!5Ü×+Ñ+¨A¨t°SÓ9ˆÜ× Ñ   D¨!Ó,ˆñ
  Ü(×5Ñ5Ø�7˜AŸD™D ´U×5EÑ5EÀqÀcÓ5J˜DÓKóˆJð Ÿ$™$Ø˜ Q§T¡T¨*¼e×>NÑ>NÐPQÈsÓ>S TÓ%TóˆKô ˜A˜{¨A¯D©D°¸7ÀDÓ,IÈ7ÓS‰Gô ×'Ñ'¨¨4°¸)ÓDÐDà—‘ØØ	�‰ˆW�gÓÜ—‘˜a˜S¬¯
©
Ô3ð ó €Iô
 �Q˜ ¨AÓ.€HÜ�Q˜˜i¨Ó-€Hà*/Ñ'€L�Ñ�{ 3Ñ'ð —T‘T˜*¬e¯l©l¸1«o�TÓ>€NØ—T‘T˜& .´w×7RÑ7R×7WÑ7W�TÓX€NØ€Hð —4‘4˜Ó(€Lô  )×;Ñ;Ø	ˆ6�8˜^¨\ÀAô Ñ€D‰/ˆ<˜1ð ×#Ñ#€JÜ×0Ñ0°Ó<ÐÜ×$Ñ$ ZÓ0€DÜ×,Ñ,¨ZÓ8€Là�o‰o˜l¨HÐ6FÓG€GØ�o‰o˜l¨HÐ6FÓG€Gä×Ñ ¨g°s¸Q±wÓ?€Gà�‰˜
¬E×,<Ñ,<¸[ÈÈ3ÐQRÉ7Ð=SÓ,TˆÓUØØ�‰˜
¬E×,<Ñ,<¸[ÈÈqÉÈÐ=SÓ,TˆÓUð€Hð
 ˆ|�‰˜xÐ=¨(Ò=¸1Ñ=€HÜ�m‰m˜L¨'°8¸TÓB€GÜ×-Ñ-Ø�g˜qŸt™t J¼×8HÑ8HÈÓ8V˜tÓWó€Gð  —?‘?Ð#3°\À7ÓK€Lð �‰Ø�¤W×%@Ñ%@×%EÑ%Eð ó €Hô 
×Ñ˜z¨8Ô4Ü	×Ñ˜z¨<Ô8à�y‰y‹{×!Ñ!Ó#€HØ�t‰tÐ(¨(¸3ˆtÓ?€HØ€Or'   zaten::diagonalc                 ó’	  — t        j                  |«      }|�|dk\  r|n||z   }|dk\  r|n||z   }t        j                  | || j	                  dt        j                  |g«      ¬«      ¬«      }t        j                  | || j	                  dt        j                  |g«      ¬«      ¬«      }| j	                  d||d¬«      }t        j                  | |d d d «      }	| j	                  d|	|¬«      }	|�Pt        t        |«      «      }
|
j                  |«       |
j                  |«       | j	                  d	||
||gz   ¬
«      }nt        j                  dd«      S | j	                  d||	«      }t        j                  | |dgd¬«      }| j	                  dt        j                  |g«      ¬«      }|dk\  r]| j	                  d| j	                  d|| j	                  d||«      «      | j	                  dt        j                  dg«      ¬«      «      }d}nZ| j	                  d| j	                  d| j	                  d||«      |«      | j	                  dt        j                  dg«      ¬«      «      }| j	                  d|d¬«      }t        j                  | |dd d «      }| j	                  d|| j	                  dt        j                  dg«      ¬«      «      }| j	                  d|| j	                  dt        j                  t        |«      dz
  g«      ¬«      «      }t        t        |«      «      d d D �cg c]?  }t        j                  | || j	                  dt        j                  |g«      ¬«      ¬«      ‘ŒA }}|j                  |«        | j                  dg|¢­ddiŽ}t        j                  | |dd d «      }| j	                  d| j	                  d|| j	                  dt        j                   dt
        j"                  ¬«      ¬«      «      «      }t%        j&                  | d|d¬«      \  }\  }}}|j	                  d||«      }t        j(                  |||dz
  g«      }|j	                  d|||dz
  ¬ «      }t        j                  ||d!d d «      }t+        j,                  |j.                  |«       t+        j,                  |j.                  |«       |S c c}w )"Nr   r?   rA   )r#   rr   r   ÚEyeLike)Úk_iÚ	Transpose)Úperm_iÚdiagonalzunknown input rankr1   rx   )Úaxes_ir0   ÚMaxÚMinr|   rC   é   ÚCumSumro   éþÿÿÿr   ÚNotr&  r@   r  r  r(  ÚGatherND)Úbatch_dims_ié   )r   r�   r¥   ra   r   rP   r  ÚzerosÚlistrS   ÚremoverÜ   r6   ÚonesÚabsrT   rQ   rî   r
   r…   rO   r	   rˆ   r†   )r   r7   ÚoffsetÚdim1Údim2ÚrankÚ	dim1_sizeÚ	dim2_sizeÚ
mask_shapeÚmaskÚaxesrÞ   Ú	offset_opÚ	diag_sizeÚselect_window_ones_fillÚselect_windowr]   Úgather_shapeÚgather_indicesÚoverrun_condÚif_opÚ
if_contextÚelse_contextr“   Úgather_indices_if_blockÚfinal_non_overrunÚfinal_overruns                              r&   rI  rI  Ó  st  € ô ×+Ñ+¨DÓ1€DàÐØ˜q’y‰t d¨T¡kˆØ˜q’y‰t d¨T¡kˆä—‘Ø	ˆ4�Q—T‘T˜*¬e×.>Ñ.>À¸vÓ.F�TÓGô€Iô —‘Ø	ˆ4�Q—T‘T˜*¬e×.>Ñ.>À¸vÓ.F�TÓGô€Ið —‘�h 	¨9¸Q�Ó?€JÜ�<‰<˜˜: t¨T°4Ó8€DØ�4‰4�	˜4 Vˆ4Ó,€Dð ÐÜ”E˜$“KÓ ˆØ�‰�DÔØ�‰�DÔØ�t‰t�K ¨d°d¸D°\Ñ.AˆtÓB‰ä×-Ñ-¨jÐ:NÓOÐOð �T‰T�%˜˜tÓ$€FÜ×.Ñ.¨q°&À"ÀÐRSÔT€Fð
 —‘�Z¬×)9Ñ)9¸6¸(Ó)C�ÓD€IØ�‚{Ø—D‘DØØ�D‰D�˜	 1§4¡4¨¨y¸)Ó#DÓEØ�D‰D�¤U×%5Ñ%5°q°cÓ%:ˆDÓ;ó
ˆ	ð
 ‰à—D‘DØØ�D‰D�˜Ÿ™˜U I¨yÓ9¸9ÓEØ�D‰D�¤U×%5Ñ%5°q°cÓ%:ˆDÓ;ó
ˆ	ð
 —‘�X˜y°�Ó3€Iô %Ÿk™k¨!¨Y¸¸4ÀÓFÐØ—D‘DØØØ	�‰ˆZ¤×!1Ñ!1°1°#Ó!6ˆÓ7ó€Mð
 —D‘DØØØ	�‰ˆZ¤×!1Ñ!1´3°v³;À±?Ð2CÓ!DˆÓEó€Mô œ˜t›Ó% c rÐ*öàô 	�‰�A�v 1§4¡4¨
¼E×<LÑ<LÈdÈVÓ<T 4Ó#UÖVð€Lð ð ×Ñ˜	Ô"Ø�1—4‘4˜Ð: <Ò:¸Ñ:€LÜ—\‘\ ! \°1°d¸DÓA€Nð —4‘4ØØ	�‰ØØØ�D‰D�¤U§\¡\°!¼5¿;¹;Ô%GˆDÓHó	
ó€Lô ,5×+GÑ+GØ	ˆ4�¨ô,Ñ(€EÑ%ˆJ˜ qð )Ÿm™m¨E°>À=ÓQÐÜ-×?Ñ?ØÐ+¨d°Q©h¨ZóÐð #Ÿ™Ø�FÐ3À$ÈÁ(ð &ó Ðô —L‘L ¨|¸QÀÀdÓK€MÜ	×Ñ˜z×/Ñ/Ð1BÔCÜ	×Ñ˜|×1Ñ1°=ÔAØ€LùòSs   Ì&ASzquantized::linearc                 ó6  — t        j                  | |«      \  }}}}t        j                  | |«      \  }	}
}}t        j                  | |||
|«      }t        j                  | |«      \  }}}}t        j                  | ||	|«      }t        j
                  | |||«      S rf   )r   Údequantize_helperÚrequantize_bias_helperr¥   ÚlinearÚquantize_helper©r   Úq_inputÚq_weightÚbiasÚop_scaleÚop_zero_pointr"   Úinput_scaler“   ÚweightÚweight_scaler]   Úq_biasr‰   s                 r&   Úquantized_linearr~  N  sœ   € ô  /×@Ñ@ÀÀGÓLÑ€Eˆ;˜˜1Ü$3×$EÑ$EÀaÈÓ$RÑ!€FˆL˜!˜TÜ×3Ñ3Ø	ˆ4�˜l¨Dó€Fô $×5Ñ5°a¸Ó@�M€Dˆ!ˆQ�ä�]‰]˜1˜e V¨TÓ2€Fä×*Ñ*¨1¨f°hÀÓNÐNr'   zquantized::linear_reluc                 ób  — t        j                  | |«      \  }}}}t        j                  | |«      \  }	}
}}t        j                  | |||
|«      }t        j                  | |«      \  }}}}t        j                  | ||	|«      }t        j
                  | |«      }t        j                  | |||«      S rf   )r   rp  rq  r¥   rr  Úrelurs  rt  s                 r&   Úquantized_linear_relur�  ^  s¬   € ô  /×@Ñ@ÀÀGÓLÑ€Eˆ;˜˜1Ü$3×$EÑ$EÀaÈÓ$RÑ!€FˆL˜!˜TÜ×3Ñ3Ø	ˆ4�˜l¨Dó€Fô $×5Ñ5°a¸Ó@�M€Dˆ!ˆQ�ä�]‰]˜1˜e V¨TÓ2€FÜ�[‰[˜˜FÓ#€Fä×*Ñ*¨1¨f°hÀÓNÐNr'   zquantized::conv1d_reluc
           
      ój  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |«      }t        j                  | |||	«      S rf   )r   rp  rq  r¥   Úconv1dr€  rs  ©r   ru  rv  rw  ÚstrideÚpaddingÚdilationÚgroupsrx  ry  r"   rz  r“   r{  r|  r]   r}  r‰   s                     r&   Úquantized_conv1d_relur‰  o  ó´   € ô  /×@Ñ@ÀÀGÓLÑ€Eˆ;˜˜1Ü$3×$EÑ$EÀaÈÓ$RÑ!€FˆL˜!˜TÜ×3Ñ3Ø	ˆ4�˜l¨Dó€Fô $×5Ñ5°a¸Ó@�M€Dˆ!ˆQ�ä�]‰]˜1˜e V¨T°6¸7ÀHÈfÓU€FÜ�[‰[˜˜FÓ#€Fä×*Ñ*¨1¨f°hÀÓNÐNr'   zquantized::conv2d_reluc
           
      ój  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |«      }t        j                  | |||	«      S rf   )r   rp  rq  r¥   Úconv2dr€  rs  r„  s                     r&   Úquantized_conv2d_relur�  ‰  rŠ  r'   zquantized::conv3d_reluc
           
      ój  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |«      }t        j                  | |||	«      S rf   )r   rp  rq  r¥   Úconv3dr€  rs  r„  s                     r&   Úquantized_conv3d_relur�  £  rŠ  r'   zquantized::conv1dc
           
      ó>  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |||	«      S rf   )r   rp  rq  r¥   rƒ  rs  r„  s                     r&   Úquantized_conv1dr’  ½  ó¤   € ô  /×@Ñ@ÀÀGÓLÑ€Eˆ;˜˜1Ü$3×$EÑ$EÀaÈÓ$RÑ!€FˆL˜!˜TÜ×3Ñ3Ø	ˆ4�˜l¨Dó€Fô $×5Ñ5°a¸Ó@�M€Dˆ!ˆQ�ä�]‰]˜1˜e V¨T°6¸7ÀHÈfÓU€Fä×*Ñ*¨1¨f°hÀÓNÐNr'   zquantized::conv2dc
           
      ó>  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |||	«      S rf   )r   rp  rq  r¥   rŒ  rs  r„  s                     r&   Úquantized_conv2dr•  Ö  r“  r'   zquantized::conv3dc
           
      ó>  — t        j                  | |«      \  }
}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | |
||||||«      }t        j
                  | |||	«      S rf   )r   rp  rq  r¥   r�  rs  r„  s                     r&   Úquantized_conv3dr—  ï  r“  r'   zquantized::conv_transpose1dc                 ó@  — t        j                  | |«      \  }}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | ||||||||«	      }t        j
                  | ||	|
«      S rf   ©r   rp  rq  r¥   Úconv_transpose2drs  ©r   ru  rv  rw  r…  r†  Úoutput_paddingr‡  rˆ  rx  ry  r"   rz  r“   r{  r|  r]   r}  r‰   s                      r&   Úquantized_conv_transpose1dr�    ó­   € ô  /×@Ñ@ÀÀGÓLÑ€Eˆ;˜˜1Ü$3×$EÑ$EÀaÈÓ$RÑ!€FˆL˜!˜TÜ×3Ñ3Ø	ˆ4�˜l¨Dó€Fô $×5Ñ5°a¸Ó@�M€Dˆ!ˆQ�ä×$Ñ$Ø	ˆ5�&˜$ ¨°ÀÈó€Fô ×*Ñ*¨1¨f°hÀÓNÐNr'   zquantized::conv_transpose2dc                 ó@  — t        j                  | |«      \  }}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | ||||||||«	      }t        j
                  | ||	|
«      S rf   r™  r›  s                      r&   Úquantized_conv_transpose2dr   $  rž  r'   zquantized::conv_transpose3dc                 ó@  — t        j                  | |«      \  }}}}t        j                  | |«      \  }}}}t        j                  | ||||«      }t        j                  | |«      \  }}}}t        j                  | ||||||||«	      }t        j
                  | ||	|
«      S rf   )r   rp  rq  r¥   Úconv_transpose3drs  r›  s                      r&   Úquantized_conv_transpose3dr£  @  rž  r'   rf   )NF)r   N)NNNr²   rÑ   )8Ú	functoolsrP   Útorch._C._onnxÚ_CÚ_onnxr‚   Ú
torch.onnxr   r   r   r   r   r}   r   r¥   r	   Útorch.onnx._internalr
   r   ÚpartialÚonnx_symbolicÚ_onnx_symbolicrä   ÚGraphContextr$   r+   r<   rd   ri   rk   rm   r�   r£   r§   r®   rÅ   rÊ   rÓ   Ú_apply_paramsrê   rü   rþ   r$  r0  rI  r~  r�  r‰  r�  r�  r’  r•  r—  r�  r   r£  rÒ   r'   r&   ú<module>r¯     s~  ðó ã ß  Ð  ÷÷ ñ ÷ 9ð #�×"Ñ" <×#=Ñ#=ÀRÔH€ñ �Ó Ø€×Ñ˜C  fÓ-ñˆy×%Ñ%ò ó .ó !ðñ Ð#Ó$Ø€×Ñ˜C  fÓ-ñ�9×)Ñ)ò ó .ó %ðñ Ð&Ó'Ø€×Ñ˜C  cÓ*ñ �i×,Ñ,ò  ó +ó (ð ñ �ÓØ€×Ñ˜C  c¨3Ó/ñ4EˆY×#Ñ#ò 4Eó 0ó ð4Eñn Ð(Ó)ñ6˜	×.Ñ.ò 6ó *ð6ñ Ð$Ó%àHLñ>Ø×Ñò>ó &ð>ñ Ð/Ó0à@DñAØ×ÑòAó 1ðAñ Ð$Ó%Ø€×Ñ˜C  c¨3Ó/àHLñEIØ×ÑòEIó 0ó &ðEIñP �ÓØ€×Ñ˜C  cÓ*ñˆi×$Ñ$ò ó +ó  ðñ( Ð%Ó&ñE�Y×+Ñ+ò Eó 'ðEñ �ÓØ€×Ñ˜C  c¨3Ó/ñ	1ˆY×#Ñ#ò 	1ó 0ó ð	1ñ Ð8Ó9Ø€×Ñ˜C  c¨3°°SÓ9ð ØñOØ×ÑòOó :ó :ðOñB Ð7Ó8Ø€×Ñ˜C  c¨3°Ó4ð Øñ!BØ×Ñò!Bó 5ó 9ð!BòH
ñ ØØ+ˆo×+Ñ+¨K¸Ó?Ð@ôñ+ó	ð+ñb Ð!Ó"ð'B�×'Ñ'ò 'Bó #ð'BñT Ð$Ó%Ø€×Ñ˜C  c¨3Ó/ñE�I×*Ñ*ò Eó 0ó &ðEñ4 �Óð30ˆI×"Ñ"ò 30ó ð30ñl Ð)Ó*àDHñIØ×ÑòIó +ðIñX Ð Ó!Ø€×Ñ˜C  c¨3Ó/ðs�	×&Ñ&ò só 0ó "ðsñr Ð#Ó$ðOØ×ÑòOó %ðOñ Ð(Ó)ðOØ×ÑòOó *ðOñ  Ð(Ó)ðOØ×ÑòOó *ðOñ2 Ð(Ó)ðOØ×ÑòOó *ðOñ2 Ð(Ó)ðOØ×ÑòOó *ðOñ2 Ð#Ó$ðOØ×ÑòOó %ðOñ0 Ð#Ó$ðOØ×ÑòOó %ðOñ0 Ð#Ó$ðOØ×ÑòOó %ðOñ0 Ð-Ó.ðOØ×ÑòOó /ðOñ6 Ð-Ó.ðOØ×ÑòOó /ðOñ6 Ð-Ó.ðOØ×ÑòOó /ñOr'   