o
    wiD                     @  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 ]8\}}}	}
t||td|tt|d d  td	|t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intmathfloorminceilr   r   r   r   nsizenoder	   
square_sumnchwr
    r3   \/home/ubuntu/sommelier/.venv/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 ]8\}}}	}
t||td|tt|d d  td	|t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+   r3   r3   r4   export_default0   s.    zLRN.export_defaultN)r   r   )__name__
__module____qualname__staticmethodr5   r7   r3   r3   r3   r4   r      s
    r   )
__future__r   r'   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r3   r3   r4   <module>   s   