a
    hY                     @   sd   d dl Z d dlmZ eedddZdee ee eeddd	Zdee ee eedd
dZdS )    N)Optional)	reductionreturnc                 C   sX   | dkrd}nF| dkrd}n8| dkr4t d d}n | dkrBd}nd	}t|  d
|S )Nnoner   mean   Zelementwise_meanzPreduction='elementwise_mean' is deprecated. Please use reduction='mean' instead.sum   z# is not a valid value for reduction)warningswarn
ValueError)r   ret r   A/var/www/auris/lib/python3.9/site-packages/torch/nn/_reduction.pyget_enum   s    r   T)size_averagereduceemit_warningr   c                 C   sP   d}| d u rd} |d u rd}| r*|r*d}n|r4d}nd}|rLt || |S )NzSsize_average and reduce args will be deprecated, please use reduction='{}' instead.Tr   r   r   )r   r   format)r   r   r   warningr   r   r   r   legacy_get_string    s    r   c                 C   s   t t| ||S )N)r   r   )r   r   r   r   r   r   legacy_get_enum7   s    r   )T)T)	r   typingr   strintr   boolr   r   r   r   r   r   <module>   s      