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 )ThresholdedReluNone)returnc                  C  s   d} t jjddgdg| d}tddddd	gtj}t|| tj}d
||| k< t	||g|gdd tj
dddtj}t|| tj}d
||| k< t	||g|gdd d S )Ng       @r   xy)inputsoutputsalphag      g        g333333?g@r   Ztest_thresholdedrelu_exampler
   r   name         Ztest_thresholdedrelu)onnxhelper	make_nodenparrayastypefloat32clipinfr   randomrandn)r   noder   r	    r   O/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/thresholdedrelu.pyexport   s       zThresholdedRelu.exportc                  C  sf   d} t jjddgdgd}tjdddtj}t|| tj	}d	||| k< t
||g|gd
d d S )Ng      ?r   r   r	   )r
   r   r   r   r   r   Ztest_thresholdedrelu_defaultr   )r   r   r   r   r   r   r   r   r   r   r   )Zdefault_alphar   r   r	   r   r   r   export_default!   s    zThresholdedRelu.export_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
   