Ë
    Q/æj�  ã                   ó’	  — d Z ddlZddlmZmZmZmZ ddlZddlmZ ddl	m
Z
mZmZ ddlmZmZ dgZ ej$                  ej&                  d¬	«      Z ed
«       ed«      dej*                  fd„«       «       Z ed«       ej.                  dddddd«      dej0                  dej0                  dej0                  dee   dee   dee   fd„«       «       Z ed ej6                  dd«      g¬«       ed ej6                  ddd¬ «      g¬«      dOd"ed#ed$efd%„«       «       Z ed&«       ej>                  d!ddd«       ej.                  ddddd'«      dej*                  dej0                  d(ee   d)ej0                  d*ej0                  d+e d,eej0                  ej0                  ej0                  f   fd-„«       «       «       Z! ed.«       ej.                  dd/«      dej*                  fd0„«       «       Z" ed1«      dPdej*                  fd2„«       Z# ed3«       ej>                  d!d!«      dej*                  fd4„«       «       Z$ ed5«      dPdej*                  fd6„«       Z% ed7«       ej>                  d!d!«      dej*                  fd8„«       «       Z& ed9«       ej>                  d!«       ej.                  ddd/«      dej*                  fd:„«       «       «       Z' ed;«       ej>                  d!«       ej.                  ddd/«      dej*                  fd<„«       «       «       Z( ed=«       ej>                  d!«       ej.                  ddd/«      dej*                  fd>„«       «       «       Z) ed?«      dej*                  fd@„«       Z* edA«       ej.                  ddd/«      dej*                  fdB„«       «       Z+ edC«       ej.                  ddddDd«      dej*                  dEej                  j0                  dFej                  j0                  dGee   dHedIej                  j0                  fdJ„«       «       Z, edK«       ej.                  dddd/d/d/dd/d/«	      dej*                  fdL„«       «       Z- edM«       ej.                  dd'ddDd«      dej*                  dEej                  j0                  dFe dGeee      dHedIej                  j0                  fdN„«       «       Z.y)Qa³  This file exports ONNX ops for opset 18.

Note [ONNX Operators that are added/updated in opset 18]

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
https://github.com/onnx/onnx/blob/main/docs/Changelog.md#version-18-of-the-default-onnx-operator-set
New operators:
    BitwiseAnd
    CenterCropPad
    Col2Im
    Mish
    OptionalGetElement
    OptionalHasElement
    Pad
    Resize
    ScatterElements
    ScatterND
    Split
é    N)ÚListÚOptionalÚSequenceÚTuple)Ú_C)Ú_type_utilsÚsymbolic_helperÚsymbolic_opset9)Ú	jit_utilsÚregistrationÚcol2imé   )Úopsetzaten::__and_zaten::bitwise_andÚgc                 ó„  — ||g}|D �cg c]  }t        j                  |«      sŒ|‘Œ }}t        |«      dk(  r|}t        j                  |Ž }t        j                  | ||«      }t        j                  | ||«      }|t
        j                  j                  k(  r| j                  d||«      S | j                  d||«      S c c}w )Nr   ÚAndÚ
BitwiseAnd)	r	   Ú_get_tensor_rankÚlenÚ_type_promote_from_valuesÚ_maybe_cast_to_typer   ÚJitScalarTypeÚBOOLÚop)r   ÚselfÚotherÚargsÚargÚ	prom_argsÚpromotion_jit_types          ú^/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/onnx/symbolic_opset18.pyÚ__and_r"   )   s¹   € ð �%ˆ=€Dà $ÖN˜¬×(HÑ(HÈÕ(M’ÐN€IÐNÜ
ˆ9ƒ~˜ÒØˆ	Ü(×BÑBÀIÐNÐÜ×.Ñ.¨q°$Ð8JÓK€DÜ×/Ñ/°°5Ð:LÓM€EØœ[×6Ñ6×;Ñ;Ò;Ø�t‰t�E˜4 Ó'Ð'Ø�4‰4�˜d EÓ*Ð*ùò Os
   ‰B=¤B=zaten::col2imÚvÚisÚinputÚoutput_sizeÚkernel_sizeÚdilationÚpaddingÚstridec           	      óè   — g }|D ]#  }t        d«      D ]  }	|j                  |«       Œ Œ% t        j                  |«      d   }
|sddg|
z  }|sdg|
z  }|sdg|
z  }| j	                  d||||||¬«      S )Né   r   é   ÚCol2Im)Údilations_iÚpads_iÚ	strides_i)ÚrangeÚappendr	   Ú_get_tensor_sizesr   )r   r%   r&   r'   r(   r)   r*   Úadjusted_paddingÚpadÚ_Únum_dimensional_axiss              r!   r   r   :   s°   € ð ÐØò )ˆÜ�q“ò 	)ˆAØ×#Ñ# CÕ(ñ	)ð)ô +×<Ñ<¸[ÓIÈ!ÑLÐÙØ˜q˜6Ð$8Ñ8ÐáØ�3Ð-Ñ-ˆáØ�Ð+Ñ+ˆà�4‰4ØØØØØØØð ó ð ó    z
aten::meanÚ
ReduceMeanÚmean)Údecoratez
aten::prodÚ
ReduceProdÚprodF)Úallow_multi_dim_supportTÚonnx_opÚnamer?   c                 ó0   — t        j                  | ||«      S ©N)r	   Ú_reduce_with_dtype_helper)r@   rA   r?   s      r!   Ú_reduce_with_dtyperE   `   s   € ô ×4Ñ4Ø�Ð.óð r9   zaten::native_layer_normÚfÚnormalized_shapeÚweightÚbiasÚepsÚreturnc                 ó6   — t        j                  | |||||«      S rC   )Úopset9Únative_layer_norm)r   r%   rG   rH   rI   rJ   s         r!   Ú_native_layer_normrO   q   s    € ô ×#Ñ# A uÐ.>ÀÈÈcÓRÐRr9   z	aten::gluÚic                 ó¾   — t        j                  ||«      }|�
|dz  dk(  sJ ‚| j                  d||dd¬«      \  }}| j                  d|| j                  d|«      «      S )Nr,   r   ÚSplit)Úaxis_iÚnum_outputs_iÚoutputsÚMulÚSigmoid)r	   Ú_get_tensor_dim_sizer   )r   r%   ÚdimÚdim_sizeÚfirstÚseconds         r!   Ú_glur]      sh   € ô ×3Ñ3°E¸3Ó?€HØÐØ˜!‰|˜qÒ Ð Ð à—D‘D˜ %°À1Èa�DÓP�M€Eˆ6Ø�4‰4��u˜aŸd™d 9¨fÓ5Ó6Ð6r9   z	aten::maxc                 ó2   — t        j                  | |||«      S rC   )r	   Ú_max_helper©r   r   Údim_or_yÚkeepdims       r!   Úmaxrc   Š   s   € ô
 ×&Ñ& q¨$°¸'ÓBÐBr9   zaten::maximumc                 ó   — t        | ||¬«      S ©N)ra   )rc   ©r   r%   r   s      r!   Úmaximumrg   ’   ó   € ô ˆq�% %Ô(Ð(r9   z	aten::minc                 ó2   — t        j                  | |||«      S rC   )r	   Ú_min_helperr`   s       r!   Úminrk   ˜   s   € ô ×&Ñ& q¨$°¸'ÓBÐBr9   zaten::minimumc                 ó   — t        | ||¬«      S re   )rk   rf   s      r!   Úminimumrm   ž   rh   r9   z
aten::amaxc                 ó˜   — | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d|||¬«      S )NÚConstant©Údtype©Úvalue_tÚ	ReduceMax©Ú
keepdims_i©r   ÚtorchÚtensorÚlong©r   r   rY   rb   Úaxess        r!   Úamaxr}   ¤   ó<   € ð �4‰4�
¤E§L¡L°¼E¿J¹JÔ$Gˆ4ÓH€DØ�4‰4�˜T 4°Gˆ4Ó<Ð<r9   z
aten::aminc                 ó˜   — | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d|||¬«      S )Nro   rp   rr   Ú	ReduceMinru   rw   r{   s        r!   Úaminr�   ¬   r~   r9   zaten::aminmaxc                 ól  — t        j                  |«      sxt        j                  |dd«      }| j                  dt	        j
                  |gt        j                  ¬«      ¬«      }| j                  d|||¬«      | j                  d|||¬«      fS | j                  d||¬«      | j                  d||¬«      fS )	NrP   rY   ro   rp   rr   r€   ru   rt   )r	   Ú_is_noneÚ
_get_constr   rx   ry   rz   r{   s        r!   Úaminmaxr…   ´   s¹   € ô ×#Ñ# CÔ(Ü×(Ñ(¨¨c°5Ó9ˆØ�t‰t�J¬¯©°c°UÄ%Ç*Á*Ô(MˆtÓNˆØ�t‰t�K  t¸ˆtÓ@À!Ç$Á$Ø˜˜t°ð CGó C
ð 
ð 	
ð �t‰t�K °'ˆtÓ:¸A¿D¹DØ˜¨'ð =Aó =
ð 
ð 	
r9   zaten::var_meanc                 ó„   — t        |«      dk(  rt        j                  | |d |d   d «      S t        j                  | |g|¢­Ž S )Nr-   r   )r   r	   Ú_var_mean_helper)r   r%   r   s      r!   Ú	_var_meanrˆ   Ä   sB   € ä
ˆ4ƒy�A‚~Ü×/Ñ/°°5¸$ÀÀQÁÈÓNÐNä×/Ñ/°°5Ð@¸4Ò@Ð@r9   zaten::logsumexpc                 óÄ   — |€| j                  d|d¬«      S | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d|||¬«      S )NÚReduceLogSumExpr   ru   ro   rp   rr   rw   )r   r%   rY   rb   r|   s        r!   Ú
_logsumexpr‹   Ì   sX   € ð €{Ø�t‰tÐ% u¸ˆtÓ;Ð;à�t‰t�J¬¯©°SÄÇ
Á
Ô(KˆtÓLˆØ�t‰tÐ% u¨d¸wˆtÓGÐGr9   zaten::linalg_matrix_normÚbr   ÚordrY   rb   rq   c                 ó6   — t        j                  | |||||«      S rC   )rM   Úlinalg_matrix_norm©r   r   r�   rY   rb   rq   s         r!   Ú_linalg_matrix_normr‘   Ö   s   € ô ×$Ñ$ Q¨¨c°3¸ÀÓGÐGr9   zaten::embedding_bagc
                 ó>   — t        j                  | |||||||||	«
      S rC   )r	   Ú_embedding_bag_helper)
r   Úembedding_matrixÚindicesÚoffsetsÚscale_grad_by_freqÚmodeÚsparseÚper_sample_weightsÚinclude_last_offsetÚpadding_idxs
             r!   Úembedding_bagr�   ã   s5   € ô ×0Ñ0Ø	ØØØØØØØØØóð r9   zaten::linalg_vector_normc                 ó6   — t        j                  | |||||«      S rC   )r	   Ú_linalg_vector_norm_helperr�   s         r!   Úlinalg_vector_normr    ÿ   s    € ô ×5Ñ5°a¸¸sÀCÈÐRWÓXÐXr9   )T)NN)/Ú__doc__Ú	functoolsÚtypingr   r   r   r   rx   r   Ú
torch.onnxr   r	   r
   rM   Útorch.onnx._internalr   r   Ú__all__ÚpartialÚonnx_symbolicÚ_onnx_symbolicÚGraphContextr"   Ú
parse_argsÚValueÚintr   Ú_apply_paramsÚstrÚboolrE   Úquantized_argsÚfloatrO   r]   rc   rg   rk   rm   r}   r�   r…   rˆ   r‹   r‘   r�   r    © r9   r!   ú<module>r´      s9  ðñó( ß 2Ó 2ã Ý ß NÑ Nß 8ð ð€ð #�×"Ñ" <×#=Ñ#=ÀRÔH€ñ �ÓÙÐ#Ó$ð+ˆi×$Ñ$ò +ó %ó  ð+ñ �ÓØ€×Ñ˜C  c¨4°°tÓ<ð!à�8‰8ð!ð —‘ð!ð —‘ð	!ð
 �s‰mð!ð �c‰]ð!ð �S‰Mò!ó =ó  ð!ñH ØÐ9˜O×9Ñ9¸,ÈÓOÐPôñ Øà%ˆ×%Ñ%Ø˜&¸%ô	
ðôñ ð ¨3ð Èò óóðñ Ð)Ó*Ø€×Ñ  e¨U°EÓ:Ø€×Ñ˜C  s¨C°Ó5ðSØ×ÑðSà�8‰8ðSð ˜s‘mðSð �H‰Hð	Sð
 �(‰(ðSð 
ðSð ˆ2�8‰8�R—X‘X˜rŸx™xÐ'Ñ(òSó 6ó ;ó +ðSñ �ÓØ€×Ñ˜C Ó%ð7ˆI×"Ñ"ò 7ó &ó ð7ñ �ÓñCˆ9×!Ñ!ò Có	 ðCñ �Ó Ø€×Ñ  dÓ+ð)ˆy×%Ñ%ò )ó ,ó !ð)ñ �ÓñCˆ9×!Ñ!ò Có ðCñ �Ó Ø€×Ñ  dÓ+ð)ˆy×%Ñ%ò )ó ,ó !ð)ñ �ÓØ€×Ñ Ó%Ø€×Ñ˜C  sÓ+ð=ˆI×"Ñ"ò =ó ,ó &ó ð=ñ
 �ÓØ€×Ñ Ó%Ø€×Ñ˜C  sÓ+ð=ˆI×"Ñ"ò =ó ,ó &ó ð=ñ
 �Ó Ø€×Ñ Ó%Ø€×Ñ˜C  cÓ*ð

ˆy×%Ñ%ò 

ó +ó &ó !ð

ñ Ð Ó!ðA�×'Ñ'ò Aó "ðAñ Ð!Ó"Ø€×Ñ˜C  sÓ+ðH�)×(Ñ(ò Hó ,ó #ðHñ Ð*Ó+Ø€×Ñ˜C  d¨C°Ó5ðHØ×ÑðHà
�(‰(�.‰.ðHð 
�‰�‰ðHð 
ˆc‰ð	Hð
 ðHð �8‰8�>‰>òHó 6ó ,ðHñ Ð%Ó&Ø€×Ñ˜C  c¨3°°S¸#¸sÀCÓHðØ×Ñòó Ió 'ðñ4 Ð*Ó+Ø€×Ñ˜C  d¨C°Ó5ðYØ×ÑðYà
�(‰(�.‰.ðYð 
ðYð 
�(˜3‘-Ñ	 ð	Yð
 ðYð �8‰8�>‰>òYó 6ó ,ñYr9   