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 )	ThresholdedRelureturnNonec                  C  s   d} t jjddgdg| d}tg dtj}t|| tj}d||| k< t	||g|gdd	 tj
d
ddtj}t|| tj}d||| k< t	||g|gdd	 d S )N       @r   xy)inputsoutputsalpha)g      g        g333333?r   g@r   test_thresholdedrelu_exampler   r   name         test_thresholdedrelu)onnxhelper	make_nodenparrayastypefloat32clipinfr   randomrandn)r   noder	   r
    r!   h/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/thresholdedrelu.pyexport   s   zThresholdedRelu.exportc                  C  sf   d} t jjddgdgd}tjdddtj}t|| tj	}d	||| k< t
||g|gd
d d S )Ng      ?r   r	   r
   )r   r   r   r   r   r   test_thresholdedrelu_defaultr   )r   r   r   r   r   r   r   r   r   r   r   )default_alphar    r	   r
   r!   r!   r"   export_default!   s   zThresholdedRelu.export_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   