U
    rh                     @  sH   d dl m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                   @  s0   e Zd ZeddddZeddddZdS )	LeakyReluNone)returnc                  C  s   t jjddgdgdd} tdddgtj}t|dtjt|tj dd  }t	| |g|gd	d
 tj
dddtj}t|dtjt|tj dd  }t	| |g|gdd
 d S )Nr   xyg?)inputsoutputsalphar      Ztest_leakyrelu_exampler
   r   name         Ztest_leakyrelu)onnxhelper	make_nodenparrayastypefloat32clipinfr   randomrandn)noder   r	    r    I/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/leakyrelu.pyexport   s       &&zLeakyRelu.exportc                  C  sp   d} t jjddgdgd}tjdddtj}t|d	tj	t|tj	 d	|   }t
||g|gd
d d S )Ng{Gz?r   r   r	   )r
   r   r   r   r   r   Ztest_leakyrelu_defaultr   )r   r   r   r   r   r   r   r   r   r   r   )Zdefault_alphar   r   r	   r    r    r!   export_leakyrelu_default   s    &z"LeakyRelu.export_leakyrelu_defaultN)__name__
__module____qualname__staticmethodr"   r#   r    r    r    r!   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r    r    r    r!   <module>   s
   