
    h                     T    S SK rS SKrS SKJr  S SKJr  S SKJr  S r	 " S S\5      r
g)    N)Base)expect)AI_ONNX_PREVIEW_TRAINING_DOMAINc                    Xb-  U-   nX-  SU-
  U-  -   nX-  SU	-
  X-  -  -   n[         R                  " U5      U
-   nS nUS:  a'  U [         R                  " SX-  -
  5      -  SX-  -
  -  nOU nX/X-  -  -
  nSU-
  U-  nUX4$ )N   r   )npsqrt)rtxgvhnorm_coefficientnorm_coefficient_postalphabetaepsilong_regularizedv_newh_newh_sqrt
r_adjustedx_newx_finals                     Z/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/adam.py
apply_adamr      s    $(1,MIUm33EHD]%BCCEWWU^g%FJ1uTW--UX>
 
en--E((E1GE      c                   <    \ rS rSr\SS j5       r\SS j5       rSrg)Adam%   Nc                     Sn SnSnSn[         R                  R                  S/ SQ/ SQU UUU[        S9n[        R
                  " S[        R                  S	9n[        R
                  " S
[        R                  S	9n[        R
                  " SS/[        R                  S	9n[        R
                  " SS/[        R                  S	9n[        R
                  " SS/[        R                  S	9n	[        R
                  " SS/[        R                  S	9n
[        XVXxXU SXU5      u  pn[        UXVXxX/XU/S[         R                  R                  [        S5      /S9  g )NMbP?ffffff?皙?gHz>r    )RTXGVH)X_newV_newH_new)inputsoutputsr   r   r   r   domaindtyper   g333333?gffffff@gGzg      g333333?g@        	test_adamr   r/   r0   nameopset_importsonnxhelper	make_noder   r   arrayfloat32int64r   r   make_opsetid)r   r   r   r   noder
   r   r   r   r   r   r   r   r   s                 r   export_adamAdam.export_adam&   s3    ! {{$$1/-2 % 	
 HHS

+HHQbhh'HHc3Zrzz2HHeT]"**5HHc3Zrzz2HHc3Zrzz2 )!.U'
e
 	!%5)(()H!L	
r   c                     Sn SnSnSn[         R                  R                  S/ SQ/ SQU UU[        S9n[        R
                  " S	[        R                  S
9n[        R
                  " S[        R                  S
9n[        R
                  " S/[        R                  S
9n[        R
                  " S/[        R                  S
9n[        R
                  " S/[        R                  S
9n	[        R
                  " S/[        R                  S
9n
[        R
                  " SS/[        R                  S
9n[        R
                  " SS/[        R                  S
9n[        R
                  " SS/[        R                  S
9n[        R
                  " SS/[        R                  S
9n[        XVXxXU SXU5      u  nnn[        XVXXU SXU5      u  nnn[        UXVX{XXX/
UUUUUU/S[         R                  R                  [        S5      /S9  g )Nr#   r$   g333333?g{Gz?r    )
r&   r'   X1X2G1G2V1V2H1H2)X1_newX2_newV1_newV2_newH1_newH2_new)r/   r0   r   r   r   r1   r%   r2   r   g      ?g      g       @g      ?g      g      @g      $@r4   test_adam_multipler   r6   r9   )r   r   r   r   rA   r
   r   x1g1v1h1x2g2v2h2x1_newv1_newh1_newx2_newv2_newh2_news                        r   export_adam_multipleAdam.export_adam_multipleR   s    !{{$$MP-2 % 
 HHS

+HHQbhh'XXse2::.XXtfBJJ/XXse2::.XXse2::.XXsCj

3XXtTl"**5XXsCj

3XXsDk4 ","""2Cg"
 ","""2Cg"

 	""""9VVVVVD%(()H!L	
r    )returnN)__name__
__module____qualname____firstlineno__staticmethodrB   rb   __static_attributes__rd   r   r   r    r    %   s)    )
 )
V 0
 0
r   r    )numpyr   r:   onnx.backend.test.case.baser   onnx.backend.test.case.noder   	onnx.defsr   r   r    rd   r   r   <module>rp      s)   
   , . 5!0^
4 ^
r   