o
    soiS                     @  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alphafloatreturnc                 C  s   | ddt | |     S )N   )npexp)r   r    r   U/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/swish.pyswish   s   r   c                   @  s   e Zd ZedddZdS )Swishr	   Nonec               	   C  s^   t jjddgdgdd} tjg dtjd}t|dd}t| |g|gd	t jd
dgd d S )Nr   r   yg      ?)inputsoutputsr   )         )dtype)r   
test_swish    )r   r   nameopset_imports)	onnxhelper	make_noder   arrayfloat32r   r   make_opsetid)noder   r   r   r   r   export   s   
zSwish.exportN)r	   r   )__name__
__module____qualname__staticmethodr%   r   r   r   r   r      s    r   )r   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   
