
    JTh                     	   S r SSKrSSKJr  SSKJr  SSKrSSKJr  SSKJ	r	J
r
Jr  SSKJrJr  S/r\R"                  " \R$                  S	S
9r\" S5      \" S5      S\R(                  4S j5       5       r\" S5      \
R,                  " SSSSSS5      S\R.                  S\R.                  S\R.                  S\\   S\\   S\\   4S j5       5       r\" S\
R4                  " SS5      /S9\" S\
R4                  " SSS S!9/S9SPS#\S$\S%\4S& jj5       5       r\" S'5      \
R<                  " S"S S S 5      \
R,                  " SSSSS(5      S\R(                  S\R.                  S)\\   S*\R.                  S+\R.                  S,\S-\ \R.                  \R.                  \R.                  4   4S. j5       5       5       r!\" S/5      \
R,                  " SS05      S\R(                  4S1 j5       5       r"\" S25      SQS\R(                  4S3 jj5       r#\" S45      \
R<                  " S"S"5      S\R(                  4S5 j5       5       r$\" S65      SQS\R(                  4S7 jj5       r%\" S85      \
R<                  " S"S"5      S\R(                  4S9 j5       5       r&\" S:5      \
R<                  " S"5      \
R,                  " SSS05      S\R(                  4S; j5       5       5       r'\" S<5      \
R<                  " S"5      \
R,                  " SSS05      S\R(                  4S= j5       5       5       r(\" S>5      \
R<                  " S"5      \
R,                  " SSS05      S\R(                  4S? j5       5       5       r)\" S@5      S\R(                  4SA j5       r*\" SB5      \
R,                  " SSS05      S\R(                  4SC j5       5       r+\" SD5      \
R,                  " SSSSES5      S\R(                  SF\R                  R.                  SG\R                  R.                  SH\,\   SI\SJ\R                  R.                  4SK j5       5       r-\" SL5      \
R,                  " SSSS0S0S0SS0S05	      S\R(                  4SM j5       5       r.\" SN5      \
R,                  " SS(SSES5      S\R(                  SF\R                  R.                  SG\SH\\\      SI\SJ\R                  R.                  4SO j5       5       r/g)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                    X/nU Vs/ s H"  n[         R                  " U5      (       d  M   UPM$     nn[        U5      S:X  a  Un[         R                  " U6 n[         R                  " XU5      n[         R                  " XU5      nU[
        R                  R                  :X  a  U R                  SX5      $ U R                  SX5      $ s  snf )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          S/var/www/auris/envauris/lib/python3.13/site-packages/torch/onnx/symbolic_opset18.py__and_r    *   s     =D $N(H(H(MIN
9~	(BBIN..q8JKD//:LME[66;;;ttE4''44d** Os
   CCzaten::col2imvisinputoutput_sizekernel_sizedilationpaddingstridec           
        ^	 / nU H'  m	UR                  U	4S j[        S5       5       5        M)     [        R                  " U5      S   nU(       d  SS/U-  nU(       d  S/U-  nU(       d  S/U-  nU R	                  SUUUUUUS9$ )Nc              3   (   >#    U  H  nTv   M	     g 7fN ).0_pads     r   	<genexpr>col2im.<locals>.<genexpr>I   s     6XXs      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     #%6U1X66  +<<[I!Lq6$883--++44       z
aten::mean
ReduceMeanmean)decoratez
aten::prod
ReduceProdprodF)allow_multi_dim_supportTonnx_opnamerC   c                 0    [         R                  " XU5      $ r+   )r   _reduce_with_dtype_helper)rD   rE   rC   s      r   _reduce_with_dtyperH   `   s     44. r=   zaten::native_layer_normfnormalized_shapeweightbiasepsreturnc                 2    [         R                  " XX#XE5      $ r+   )opset9native_layer_norm)r   r#   rJ   rK   rL   rM   s         r   _native_layer_normrR   q   s     ##A.>RRr=   z	aten::gluic                     [         R                  " X5      nUb  US-  S:X  d   eU R                  SXSSS9u  pEU R                  SX@R                  SU5      5      $ )Nr2   r   Split)axis_inum_outputs_ioutputsMulSigmoid)r   _get_tensor_dim_sizer   )r   r#   dimdim_sizefirstseconds         r   _glur`      s`     33E?H!|q   DD%1aDPME44udd9f566r=   z	aten::maxc                 0    [         R                  " XX#5      $ r+   )r   _max_helperr   r   dim_or_ykeepdims       r   maxrf      s    
 &&qBBr=   zaten::maximumc                     [        XUS9$ N)rd   )rf   r   r#   r   s      r   maximumrj           q%((r=   z	aten::minc                 0    [         R                  " XX#5      $ r+   )r   _min_helperrc   s       r   minrn      s     &&qBBr=   zaten::minimumc                     [        XUS9$ rh   )rn   ri   s      r   minimumrp      rk   r=   z
aten::amaxc                     U R                  S[        R                  " U[        R                  S9S9nU R                  SXUS9$ )NConstantdtypevalue_t	ReduceMax
keepdims_ir   torchtensorlongr   r   r\   re   axess        r   amaxr      :     44
ELLEJJ$G4HD44TG4<<r=   z
aten::aminc                     U R                  S[        R                  " U[        R                  S9S9nU R                  SXUS9$ )Nrr   rs   ru   	ReduceMinrx   rz   r~   s        r   aminr      r   r=   zaten::aminmaxc                 P   [         R                  " U5      (       dl  [         R                  " USS5      nU R                  S[        R
                  " U/[        R                  S9S9nU R                  SXUS9U R                  SXUS94$ U R                  SXS9U R                  SXS94$ )	NrS   r\   rr   rs   ru   r   rx   rw   )r   _is_none
_get_constr   r{   r|   r}   r~   s        r   aminmaxr      s     ##C((((c59ttJcU%**(MtNttKt@!$$ CG C
 
 	
 ttKt:ADD =A =
 
 	
r=   zaten::var_meanc                     [        U5      S:X  a  [        R                  " XS US   S 5      $ [        R                  " X/UQ76 $ )Nr3   r   )r   r   _var_mean_helper)r   r#   r   s      r   	_var_meanr      s>    
4yA~//$QNN//@4@@r=   zaten::logsumexpc                     Uc  U R                  SUSS9$ U R                  S[        R                  " U[        R                  S9S9nU R                  SXUS9$ )NReduceLogSumExpr   rx   rr   rs   ru   rz   )r   r#   r\   re   r   s        r   
_logsumexpr      sV     {tt%ut;;ttJS

(KtLtt%uwtGGr=   zaten::linalg_matrix_normbr   ordr\   re   rt   c                 2    [         R                  " XX#XE5      $ r+   )rP   linalg_matrix_normr   r   r   r\   re   rt   s         r   _linalg_matrix_normr      s     $$QcGGr=   zaten::embedding_bagc
                 @    [         R                  " U UUUUUUUUU	5
      $ 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                 2    [         R                  " XX#XE5      $ r+   )r   _linalg_vector_norm_helperr   s         r   linalg_vector_normr      s     55asXXr=   )T)NN)0__doc__	functoolscollections.abcr   typingr   r{   r   
torch.onnxr   r   r   rP   torch.onnx._internalr	   r
   __all__partialonnx_symbolic_onnx_symbolicGraphContextr    
parse_argsValueintr   _apply_paramsstrboolrH   quantized_argsfloattuplerR   r`   rf   rj   rn   rp   r   r   r   r   r   listr   r   r   r,   r=   r   <module>r      s  (  $    N N 8  ""<#=#=RH #$+i$$ + %  + Cc4t< 88    	 
 sm  c]  SM  =   F O99,OP %%&%	
 3   )*eUE:CsC5SS88S smS HH	S
 ((S 
S 288RXXrxx'(S 6 ; +S C%7I"" 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dC5HH
((..H 
H 
c	H
 H 88>>H 6 ,H %&Cc3S#sCH I '4 *+CdC5YY
((..Y 
Y 
(3-	 	Y
 Y 88>>Y 6 ,Yr=   