
    Ng!                     R    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 )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )LRNreturnNc                     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          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           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intmathfloorminceilr   r   r   r   nsizenoder
   
square_sumnchwr   s               [/var/www/html/ai-engine/env/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%(1s4:uqyAo#>#>???@@31s49eai1_#=#=>>>BD D   
& 
&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          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           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               r4   export_defaultzLRN.export_default/   s   {$$UC53%a$PPIOOAq!Q''..rz::Xl++222:>>
*QW-- 	 	JAq!Q%(1s4:uqyAo#>#>???@@31s49eai1_#=#=>>>BD D   
& 
&Jq!Qz"" $%%-:55$>?tQC!3EFFFFFFr6   )r   N)__name__
__module____qualname__staticmethodr5   r9    r6   r4   r   r      sX        ? ? ? \?> G G G \G G Gr6   r   )	r(   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r>   r6   r4   <module>rB      s   
       , , , , , , . . . . . .7G 7G 7G 7G 7G$ 7G 7G 7G 7G 7Gr6   