U
    rh                     @  sX   d dl mZ d dlZd dlZd dlmZ d dlmZ dddddZ	G d	d
 d
eZ
dS )    )annotationsN)Base)expectz
np.ndarray)xreturnc              	   C  s,   t d}d}| tdtd||  |  S )NgUUUUUU?g      ?r      )floatnpmaximumZminimum)r   Zalfabeta r   I/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/hardswish.py	hardswish   s    r   c                   @  s   e Zd ZeddddZdS )	HardSwishNone)r   c                  C  sN   t jjddgdgd} tjdddtj}t|}t	| |g|gdd	 d S )
Nr   r   y)inputsoutputs         Ztest_hardswish)r   r   name)
onnxhelperZ	make_noder	   randomZrandnZastypeZfloat32r   r   )noder   r   r   r   r   export   s    zHardSwish.exportN)__name__
__module____qualname__staticmethodr   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   r   <module>   s   