
    Si$                    `    S SK Jr  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)	    )annotationsN)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                     _/var/www/html/ai-image-ml/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(   c                    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   r0   r1   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)r0   r1   r   r   r   r2   r&   r3   r   g      ?g      g       @g      ?g      g      @g      $@r5   test_adam_multipler   r7   r:   )r   r   r   r   rB   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_multipleU   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    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodrC   rc   __static_attributes__re   r   r   r!   r!   (   s)    )
 )
V 0
 0
r   r!   )
__future__r   numpyr	   r;   onnx.backend.test.case.baser   onnx.backend.test.case.noder   	onnx.defsr   r   r!   re   r   r   <module>rs      s,    #   , . 5!4^
4 ^
r   