
    /j                    V    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e          Z	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )ThresholdedRelureturnNonec                 .   d} t           j                            ddgdg|           }t          j        g d                              t          j                  }t          j        || t          j                  }d||| k    <   t          ||g|gd	           t          j
                            d
dd                              t          j                  }t          j        || t          j                  }d||| k    <   t          ||g|g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   s       p/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/thresholdedrelu.pyexportzThresholdedRelu.export   s   {$$secU% % 
 
 H///0077
CCGAubf%%!u*tQC!3QRRRRIOOAq!$$++BJ77GAubf%%!u*tQC!3IJJJJJJ    c                 J   d} t           j                            ddgdg          }t          j                            ddd                              t          j                  }t          j        || t          j	                  }d	||| k    <   t          ||g|g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   s       r$   export_defaultzThresholdedRelu.export_default!   s    {$$%6use$TTIOOAq!$$++BJ77GA}bf-- !!}
tQC!3QRRRRRRr&   N)r   r	   )__name__
__module____qualname__staticmethodr%   r*    r&   r$   r   r      s\        K K K \K$ S S S \S S Sr&   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r/   r&   r$   <module>r4      s    # " " " " "      , , , , , , . . . . . .S S S S Sd S S S S Sr&   