
    bIgD                    V    d dl mZ d dlZd dlZd dlZd dlmZ d dlm	Z	  G d de      Z
y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)LRNc                    d} d}d}d}t         j                  j                  ddgdg| |||      }t        j                  j                  d	d	d	d	      j                  t        j                        }t        j                  d
      j                  t        j                        }t        j                  |j                        D ]  \  }}}	}
t        ||t        d|t        t        j                  |dz
  dz              z
        t!        d	|t        t        j"                  |dz
  dz              z   dz         |	|
f   dz        ||||	|
f<    ||| |z  |z  z   |z  z  }t%        ||g|gd       y )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               \/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/lrn.pyexportz
LRN.export   ss   {{$$5E % 
 IIOOAq!Q'..rzz:XXl+222::>
**QWW- 	JAq!Q%(1s4::uqyAo#>??@31s499eai1_#=>>BD   
&Jq!Qz"	 $%%-:55$>?tQC!:>    c                    d} d}d}d}t         j                  j                  ddgdgd      }t        j                  j                  d	d	d	d	      j                  t        j                        }t        j                  d
      j                  t        j                        }t        j                  |j                        D ]  \  }}}	}
t        ||t        d|t        t        j                  |dz
  dz              z
        t!        d	|t        t        j"                  |dz
  dz              z   dz         |	|
f   dz        ||||	|
f<    ||| |z  |z  z   |z  z  }t%        ||g|gd       y )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_defaultzLRN.export_default0   sa   {{$$UC53%a$PIIOOAq!Q'..rzz:XXl+222::>
**QWW- 	JAq!Q%(1s4::uqyAo#>??@31s499eai1_#=>>BD   
&Jq!Qz"	 $%%-:55$>?tQC!3EFr6   N)returnNone)__name__
__module____qualname__staticmethodr5   r9    r6   r4   r   r      s*    ? ?> G Gr6   r   )
__future__r   r(   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r@   r6   r4   <module>rE      s'    #    , .7G$ 7Gr6   