o
    wÖii  ã                   @  sN   d dl mZ d dlmZ d dlmZ erd dlZdd
d„ZG dd„ deƒZ	dS )é    )Úannotations)ÚTYPE_CHECKING)ÚOpRunUnaryNumNÚxú
np.ndarrayÚalphaÚfloatÚreturnc                 C  s0   | dk  | j¡}||d |   | j¡8 }| | S )Nr   é   )ÚastypeÚdtype)r   r   Úsign© r   ú]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_leaky_relu.pyÚ_leaky_relu   s   r   c                   @  s   e Zd Zddd„ZdS )Ú	LeakyReluNc                 C  s   |p| j }t||ƒ |j¡fS ©N)r   r   r   r   )Úselfr   r   r   r   r   Ú_run   s   
zLeakyRelu._runr   )Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s    r   )r   r   r   r   r	   r   )
Ú
__future__r   Útypingr   Úonnx.reference.ops._opr   ÚnumpyÚnpr   r   r   r   r   r   Ú<module>   s   
