
    ΧgY              	           d dl Z d dlmZ dedefdZ	 ddee         dee         d	edefd
Z	 ddee         dee         d	edefdZdS )    N)Optional	reductionreturnc                     | dk    rd}nC| dk    rd}n:| dk    rt          j        d           d}n| dk    rd}nd	}t          |  d
          |S )Nnoner   mean   elementwise_meanzPreduction='elementwise_mean' is deprecated. Please use reduction='mean' instead.sum   z# is not a valid value for reduction)warningswarn
ValueError)r   rets     O/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/nn/_reduction.pyget_enumr      s    F	f			(	(	(3	
 	
 	
 	e		IJJJKKKJ    Tsize_averagereduceemit_warningc                     d}| d} |d}| r|rd}n|rd}nd}|r't          j        |                    |                     |S )NzSsize_average and reduce args will be deprecated, please use reduction='{}' instead.Tr   r   r   )r   r   format)r   r   r   warningr   s        r   legacy_get_stringr       sr    
 dG~  	  +gnnS))***Jr   c                 >    t          t          | ||                    S )N)r   r   )r   r   r   s      r   legacy_get_enumr   7   s    
 %lFLIIJJJr   )T)	r   typingr   strintr   boolr   r    r   r   <module>r#      s               6  4.TN  		   4 K K4.KTNK K 		K K K K K Kr   