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 )HardSigmoidNone)returnc                  C  s   t jjddgdgddd} tddd	gtj}t|d d dd	}t| |g|gd
d tj	
dddtj}t|d d dd	}t| |g|gdd d S )Nr   xy      ?g333333?)inputsoutputsalphabetar      Ztest_hardsigmoid_exampler   r   name         Ztest_hardsigmoid)onnxhelper	make_nodenparrayastypefloat32clipr   randomrandn)noder   r	    r!   K/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/hardsigmoid.pyexport   s        zHardSigmoid.exportc                  C  sd   d} d}t jjddgdgd}tjddd	tj}t||  | d
d}t	||g|gdd d S )Ng?r
   r   r   r	   )r   r   r   r   r   r   r   Ztest_hardsigmoid_defaultr   )
r   r   r   r   r   r   r   r   r   r   )Zdefault_alphaZdefault_betar    r   r	   r!   r!   r"   export_hardsigmoid_default   s    z&HardSigmoid.export_hardsigmoid_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
   