U
    q±Ëh1  ã                   @  sF   d dl mZ d dlZd dlmZ ddddœdd„ZG d	d
„ d
eƒZdS )é    )ÚannotationsN)ÚOpRunUnaryNumz
np.ndarrayÚfloat)ÚxÚalphaÚreturnc                 C  s0   | dk  | j¡}||d |   | j¡8 }| | S )Nr   é   )ÚastypeÚdtype)r   r   Úsign© r   úD/tmp/pip-unpacked-wheel-xnis5xre/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._run)N)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   )Ú
__future__r   ZnumpyÚnpZonnx.reference.ops._opr   r   r   r   r   r   r   Ú<module>   s   