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 )		LeakyRelureturnNonec                  C  s   t jjddgdgdd} tg dtj}t|dtjt|tj dd  }t	| |g|gdd	 tj
d
ddtj}t|dtjt|tj dd  }t	| |g|gdd	 d S )Nr   xyg?)inputsoutputsalpha)r      r   test_leakyrelu_exampler
   r   name         test_leakyrelu)onnxhelper	make_nodenparrayastypefloat32clipinfr   randomrandn)noder   r	    r"   b/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/leakyrelu.pyexport   s   &&zLeakyRelu.exportc                  C  sp   d} t jjddgdgd}tjdddtj}t|d	tj	t|tj	 d	|   }t
||g|gd
d d S )Ng{Gz?r   r   r	   )r
   r   r   r   r   r   test_leakyrelu_defaultr   )r   r   r   r   r   r    r   r   r   r   r   )default_alphar!   r   r	   r"   r"   r#   export_leakyrelu_default   s   &z"LeakyRelu.export_leakyrelu_defaultN)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   