Ë
    [^(hœ  ã                   ó’	  — d Z ddlZddlmZ ddl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j.                  dej.                  dej.                  dee   dee   dee   fd„«       «       Z ed e
j4                  dd«      g¬«       ed e
j4                  ddd ¬!«      g¬«      dPd#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j.                  d)ee   d*ej.                  d+ej.                  d,ed-e ej.                  ej.                  ej.                  f   fd.„«       «       «       Z! ed/«       e
j,                  dd0«      dej(                  fd1„«       «       Z" ed2«      dQdej(                  fd3„«       Z# ed4«       e
j<                  d"d"«      dej(                  fd5„«       «       Z$ ed6«      dQdej(                  fd7„«       Z% ed8«       e
j<                  d"d"«      dej(                  fd9„«       «       Z& ed:«       e
j<                  d"«       e
j,                  ddd0«      dej(                  fd;„«       «       «       Z' ed<«       e
j<                  d"«       e
j,                  ddd0«      dej(                  fd=„«       «       «       Z( ed>«       e
j<                  d"«       e
j,                  ddd0«      dej(                  fd?„«       «       «       Z) ed@«      dej(                  fdA„«       Z* edB«       e
j,                  ddd0«      dej(                  fdC„«       «       Z+ edD«       e
j,                  ddddEd«      dej(                  dFej                  j.                  dGej                  j.                  dHe,e   dIedJej                  j.                  fdK„«       «       Z- edL«       e
j,                  dddd0d0d0dd0d0«	      dej(                  fdM„«       «       Z. edN«       e
j,                  dd(ddEd«      dej(                  dFej                  j.                  dGedHeee      dIedJej                  j.                  fdO„«       «       Z/y)Ra³  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)ÚSequence)ÚOptional)Ú_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          úY/var/www/skyplay_api_hub/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 ]%  Š	|j                  ˆ	fd„t        d«      D «       «       Œ' t        j                  |«      d   }|sddg|z  }|sdg|z  }|sdg|z  }| j	                  d||||||¬«      S )Nc              3   ó"   •K  — | ]  }‰–— Œ y ­w©N© )Ú.0Ú_Úpads     €r   ú	<genexpr>zcol2im.<locals>.<genexpr>I   s   øè ø€ Ò6¨¤Ñ6ùs   ƒé   r   é   ÚCol2Im)Údilations_iÚpads_iÚ	strides_i)ÚextendÚranger   Ú_get_tensor_sizesr   )
r   r#   r$   r%   r&   r'   r(   Úadjusted_paddingÚnum_dimensional_axisr/   s
            @r   r   r   ;   s¨   ø€ ð #%ÐØò 7ˆØ×ÑÓ6¬U°1«XÔ6Õ6ð7ô +×<Ñ<¸[Ó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ÚnamerB   c                 ó0   — t        j                  | ||«      S r+   )r   Ú_reduce_with_dtype_helper)rC   rD   rB   s      r   Ú_reduce_with_dtyperG   `   s   € ô ×4Ñ4Ø�Ð.óð r<   zaten::native_layer_normÚfÚnormalized_shapeÚweightÚbiasÚepsÚreturnc                 ó6   — t        j                  | |||||«      S r+   )Úopset9Únative_layer_norm)r   r#   rI   rJ   rK   rL   s         r   Ú_native_layer_normrQ   q   s    € ô ×#Ñ# A uÐ.>ÀÈÈcÓRÐRr<   z	aten::gluÚic                 ó¾   — t        j                  ||«      }|�
|dz  dk(  sJ ‚| j                  d||dd¬«      \  }}| j                  d|| j                  d|«      «      S )Nr1   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Ð6r<   z	aten::maxc                 ó2   — t        j                  | |||«      S r+   )r   Ú_max_helper©r   r   Údim_or_yÚkeepdims       r   Úmaxre   Š   s   € ô
 ×&Ñ& q¨$°¸'ÓBÐBr<   zaten::maximumc                 ó   — t        | ||¬«      S ©N)rc   )re   ©r   r#   r   s      r   Úmaximumri   ’   ó   € ô ˆq�% %Ô(Ð(r<   z	aten::minc                 ó2   — t        j                  | |||«      S r+   )r   Ú_min_helperrb   s       r   Úminrm   ˜   s   € ô ×&Ñ& q¨$°¸'ÓBÐBr<   zaten::minimumc                 ó   — t        | ||¬«      S rg   )rm   rh   s      r   Úminimumro   ž   rj   r<   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   r[   rd   Úaxess        r   Úamaxr   ¤   ó<   € ð �4‰4�
¤E§L¡L°¼E¿J¹JÔ$Gˆ4ÓH€DØ�4‰4�˜T 4°Gˆ4Ó<Ð<r<   z
aten::aminc                 ó˜   — | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d|||¬«      S )Nrq   rr   rt   Ú	ReduceMinrw   ry   r}   s        r   Úaminrƒ   ¬   r€   r<   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 )	NrR   r[   rq   rr   rt   r‚   rw   rv   )r   Ú_is_noneÚ
_get_constr   rz   r{   r|   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ó =
ð 
ð 	
r<   zaten::var_meanc                 ó„   — t        |«      dk(  rt        j                  | |d |d   d «      S t        j                  | |g|¢­Ž S )Nr2   r   )r   r   Ú_var_mean_helper)r   r#   r   s      r   Ú	_var_meanrŠ   Ä   sB   € ä
ˆ4ƒy�A‚~Ü×/Ñ/°°5¸$ÀÀQÁÈÓNÐNä×/Ñ/°°5Ð@¸4Ò@Ð@r<   zaten::logsumexpc                 óÄ   — |€| j                  d|d¬«      S | j                  dt        j                  |t        j                  ¬«      ¬«      }| j                  d|||¬«      S )NÚReduceLogSumExpr   rw   rq   rr   rt   ry   )r   r#   r[   rd   r~   s        r   Ú
_logsumexpr�   Ì   sX   € ð €{Ø�t‰tÐ% u¸ˆtÓ;Ð;à�t‰t�J¬¯©°SÄÇ
Á
Ô(KˆtÓLˆØ�t‰tÐ% u¨d¸wˆtÓGÐGr<   zaten::linalg_matrix_normÚbr   Úordr[   rd   rs   c                 ó6   — t        j                  | |||||«      S r+   )rO   Úlinalg_matrix_norm©r   r   r�   r[   rd   rs   s         r   Ú_linalg_matrix_normr“   Ö   s   € ô ×$Ñ$ Q¨¨c°3¸ÀÓGÐGr<   zaten::embedding_bagc
                 ó>   — t        j                  | |||||||||	«
      S r+   )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Ø	ØØØØØØØØØóð r<   zaten::linalg_vector_normc                 ó6   — t        j                  | |||||«      S r+   )r   Ú_linalg_vector_norm_helperr’   s         r   Úlinalg_vector_normr¢   ÿ   s    € ô ×5Ñ5°a¸¸sÀCÈÐRWÓXÐXr<   )T)NN)0Ú__doc__Ú	functoolsÚcollections.abcr   Útypingr   rz   r   Ú
torch.onnxr   r   r   rO   Útorch.onnx._internalr	   r
   Ú__all__ÚpartialÚonnx_symbolicÚ_onnx_symbolicÚGraphContextr    Ú
parse_argsÚValueÚintr   Ú_apply_paramsÚstrÚboolrG   Úquantized_argsÚfloatÚtuplerQ   r_   re   ri   rm   ro   r   rƒ   r‡   rŠ   r�   Úlistr“   rŸ   r¢   r,   r<   r   ú<module>r¸      s9  ðñó( Ý $Ý ã Ý ß NÑ Nß 8ð ð€ð #�×"Ñ" <×#=Ñ#=ÀRÔH€ñ �ÓÙÐ#Ó$ð+ˆi×$Ñ$ò +ó %ó  ð+ñ �ÓØ€×Ñ˜C  c¨4°°tÓ<ð à�8‰8ð ð —‘ð ð —‘ð	 ð
 �s‰mð ð �c‰]ð ð �S‰Mò ó =ó  ð ñF ØÐ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r<   