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 )Elu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|tj dd d  }t
| |g|gd	d
 tjdddtj}t|dtjt	t|tj dd d  }t
| |g|gdd
 d S )Nr   xyg       @)inputsoutputsalphar      Ztest_elu_exampler
   r   name         Ztest_elu)onnxhelper	make_nodenparrayastypefloat32clipinfexpr   randomrandn)noder   r	    r!   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/elu.pyexport   s    00z
Elu.exportc                  C  sz   d} t jjddgdgd}tjdddtj}t|d	tj	t
t|tj	 d	d
 |   }t||g|gdd d S )Ng      ?r   r   r	   )r
   r   r   r   r   r   r   Ztest_elu_defaultr   )r   r   r   r   r   r   r   r   r   r   r   r   )Zdefault_alphar    r   r	   r!   r!   r"   export_elu_default   s    0zElu.export_elu_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
   