o
    soi0                     @  sP   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 deZ
dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	LRNreturnNonec                  C  s   d} d}d}d}t jjddgdg| |||d}tjd	d	d	d	tj}td
tj}t	|j
D ]4\}}}	}
t||td|t|d d  td	|t|d d  d |	|
f d ||||	|
f< q3||| | |  |  }t||g|gd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
    r2   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/lrn.pyexport   s>   	z
LRN.exportc                  C  s   d} d}d}d}t jjddgdgdd}tjd	d	d	d	tj}td
tj}t	|j
D ]4\}}}	}
t||td|t|d d  td	|t|d d  d |	|
f d ||||	|
f< q0||| | |  |  }t||g|gd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*   r2   r2   r3   export_default0   s.   zLRN.export_defaultN)r   r   )__name__
__module____qualname__staticmethodr4   r6   r2   r2   r2   r3   r      s
    r   )
__future__r   r&   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r2   r2   r2   r3   <module>   s   