o
    wi                     @  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                   @  s(   e Zd ZedddZedddZdS )	PRelureturnNonec                  C  s   t jjdddgdgd} tjdddtj}tjdddtj}t|d	tj	t|tj	 d	|  }t
| ||g|gd
d d S )Nr   xslopeyinputsoutputs         r   test_prelu_exampler   r   nameonnxhelper	make_nodenprandomrandnastypefloat32clipinfr   noder   r	   r
    r!   ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/prelu.pyexport   s   &zPRelu.exportc                  C  s   t jjdddgdgd} tjdddtj}tjdtj}t|d	tj	t|tj	 d	|  }t
| ||g|gd
d d S )Nr   r   r	   r
   r   r   r   r   r   test_prelu_broadcastr   r   r   r!   r!   r"   export_prelu_broadcast   s   &zPRelu.export_prelu_broadcastN)r   r   )__name__
__module____qualname__staticmethodr#   r%   r!   r!   r!   r"   r      s
    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r!   r!   r!   r"   <module>   s   