
    /j0                    ^    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
dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )LRNreturnNonec                    d} d}d}d}t           j                            ddgdg| |||          }t          j                            d	d	d	d	                              t          j                  }t          j        d
                              t          j                  }t          j	        |j
                  D ]\  }}}	}
t          ||t          d|t          j        |dz
  dz            z
            t          d	|t          j        |dz
  dz            z   dz             |	|
f         dz            ||||	|
f<   ||| |z  |z  z   |z  z  }t#          ||g|gd           d S )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mathfloorminceilr   r   r   r   nsizenoder   
square_sumnchwr   s               d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/lrn.pyexportz
LRN.export   s   {$$5E % 
 
 IOOAq!Q''..rz::Xl++222:>>
*QW-- 	 	JAq!Q%(1tz519/:::;;c1ty%!)q999A=? ?   
& 
&Jq!Qz"" $%%-:55$>?tQC!:>>>>>>    c                    d} d}d}d}t           j                            ddgdgd          }t          j                            d	d	d	d	                              t          j                  }t          j        d
                              t          j                  }t          j	        |j
                  D ]\  }}}	}
t          ||t          d|t          j        |dz
  dz            z
            t          d	|t          j        |dz
  dz            z   dz             |	|
f         dz            ||||	|
f<   ||| |z  |z  z   |z  z  }t#          ||g|gd           d S )Ng-C6?g      ?g      ?r   r   r   r   )r   r   r   r   r   r   r   r   test_lrn_defaultr   r   r-   s               r5   export_defaultzLRN.export_default0   s{   {$$UC53%a$PPIOOAq!Q''..rz::Xl++222:>>
*QW-- 	 	JAq!Q%(1tz519/:::;;c1ty%!)q999A=? ?   
& 
&Jq!Qz"" $%%-:55$>?tQC!3EFFFFFFr7   N)r   r	   )__name__
__module____qualname__staticmethodr6   r:    r7   r5   r   r      sX        ? ? ? \?> G G G \G G Gr7   r   )
__future__r   r)   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r?   r7   r5   <module>rD      s    # " " " " "       , , , , , , . . . . . .7G 7G 7G 7G 7G$ 7G 7G 7G 7G 7Gr7   