
    h!                     J    S SK r S SKrS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   <    \ rS rSr\SS j5       r\SS j5       rSrg)LRN   Nc                     Sn SnSnSn[         R                  R                  SS/S/U UUUS9n[        R                  R                  S	S	S	S	5      R                  [        R                  5      n[        R                  " S
5      R                  [        R                  5      n[        R                  " UR                  5       H  u  pxp[        UU[        SU[        [        R                  " US-
  S-  5      5      -
  5      [!        S	U[        [        R"                  " US-
  S-  5      5      -   S-   5      2U	U
4   S-  5      XgXU
4'   M     XRX-  U-  -   U-  -  n[%        XE/U/SS9  g )Ng-C6*?g      ?g       @   r   xy)inputsoutputsalphabetabiassize   r   r   r   r   r         test_lrnr   r   nameonnxhelper	make_nodenprandomrandnastypefloat32zerosndindexshapesummaxintmathfloorminceilr   r   r   r   nsizenoder
   
square_sumnchwr   s               Y/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/lrn.pyexport
LRN.export   s`   {{$$5E % 
 IIOOAq!Q'..rzz:XXl+222::>
**QWW-JA!%(1s4::uqyAo#>??@31s499eai1_#=>>BD   
&J!z" . %-:55$>?tC!:>    c                     Sn SnSnSn[         R                  R                  SS/S/SS9n[        R                  R                  S	S	S	S	5      R                  [        R                  5      n[        R                  " S
5      R                  [        R                  5      n[        R                  " UR                  5       H  u  pxp[        UU[        SU[        [        R                  " US-
  S-  5      5      -
  5      [!        S	U[        [        R"                  " US-
  S-  5      5      -   S-   5      2U	U
4   S-  5      XgXU
4'   M     XRX-  U-  -   U-  -  n[%        XE/U/SS9  g )Ng-C6?g      ?g      ?r	   r   r
   r   )r   r   r   r   r   r   r   r   test_lrn_defaultr   r   r,   s               r4   export_defaultLRN.export_default/   sN   {{$$UC53%a$PIIOOAq!Q'..rzz:XXl+222::>
**QWW-JA!%(1s4::uqyAo#>??@31s499eai1_#=>>BD   
&J!z" . %-:55$>?tC!3EFr7    )returnN)__name__
__module____qualname____firstlineno__staticmethodr5   r:   __static_attributes__r<   r7   r4   r   r      s*    ? ?> G Gr7   r   )	r(   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r<   r7   r4   <module>rG      s$   
    , .7G$ 7Gr7   