
    JTh	                     R   S r SSKrSSKJs  Jr  SSKJr  SSKJ	r	  SSK
JrJr  / SQrS r\R                  " \R                   SS	9r\" S
5      \	R$                  " SSSSS5      S\R&                  S\R(                  S\R(                  S\S\S\4S j5       5       r\" S5      \	R$                  " SSS5      S\R&                  S\R(                  S\R(                  S\4S j5       5       r\" S5      \	R$                  " SS5      SS\R&                  S\R(                  S\4S jj5       5       rg)a  This file exports ONNX ops for opset 20.

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

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
https://github.com/onnx/onnx/blob/main/docs/Changelog.md#version-20-of-the-default-onnx-operator-set
New operators:
    AffineGrid
    ConstantOfShape
    DFT
    Gelu
    GridSample
    ImageDecoder
    IsInf
    IsNaN
    ReduceMax
    ReduceMin
    RegexFullMatch
    StringConcat
    StringSplit
    N)_C)symbolic_helper)	jit_utilsregistration)_grid_sampler_affine_grid_generatorgeluc                 &    U S:X  a  S$ U S:X  a  S$ U $ )Nbilinearlinearbicubiccubic )mode_ss    S/var/www/auris/envauris/lib/python3.13/site-packages/torch/onnx/symbolic_opset20.pyconvert_grid_sample_moder   &   s(    j(9?99LgRX       )opsetzaten::grid_samplervibginputgrid	mode_enumpadding_mode_enumalign_cornersc           	      B   [         R                  R                  5        VVs0 s H  u  pgXv_M	     snnU   n[        U5      n[         R                  R                  5        VVs0 s H  u  pgXv_M	     snnU   n	U R                  SUU[        U5      UU	S9$ s  snnf s  snnf )N
GridSample)align_corners_ir   padding_mode_s)FGRID_SAMPLE_INTERPOLATION_MODESitemsr   GRID_SAMPLE_PADDING_MODESopint)
r   r   r   r   r   r   kr   r   r"   s
             r   r   r   /   s      !@@FFHIHtqadHI)TF%f-F'('B'B'H'H'JK'Jtqad'JKN 44M*%    J Ls   B#Bzaten::affine_grid_generatorthetasizec                 8    U R                  SUU[        U5      S9$ )N
AffineGrid)r!   )r'   r(   )r   r*   r+   r   s       r   r   r   I   s+     44M*	   r   z
aten::gelusselfapproximatec                 "    U R                  SXS9$ )NGelu)approximate_s)r'   )r   r/   r0   s      r   r	   r	   Y   s     44488r   )none)__doc__	functoolstorch.nn.functionalnn
functionalr#   torchr   
torch.onnxr   torch.onnx._internalr   r   __all__r   partialonnx_symbolic_onnx_symbolic
parse_argsGraphContextValuer(   boolr   r   strr	   r   r   r   <module>rF      so  ,     & 8 > ""<#=#=RH $%Cc3488 (( 	
   5 &0 -.Cc*88 (( 	 + / C%9I"" 9"(( 9 9 & 9r   