o
    soi                     @  sR   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	Z	G d
d deZ
dS )    )annotationsN)Base)expectx
np.ndarrayreturnc              	   C  s,   t d}d}| tdtd||  |  S )NgUUUUUU?g      ?r      )floatnpmaximumminimum)r   alfabeta r   Y/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/hardswish.py	hardswish   s    r   c                   @  s   e Zd ZedddZdS )	HardSwishr   Nonec                  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         test_hardswish)r   r   name)
onnxhelper	make_noder
   randomrandnastypefloat32r   r   )noder   r   r   r   r   export   s   zHardSwish.exportN)r   r   )__name__
__module____qualname__staticmethodr$   r   r   r   r   r      s    r   )r   r   r   r   )
__future__r   numpyr
   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r   r   r   <module>   s   
