a
    ¢º”hö  ã                   @   s–   U d dl Z d dlZd dl mZ d dlmZmZ d dlmZ e  ¡ Zee	e
f ed< edƒZedƒZe	ddœd	d
„Zeeef eeef dœdd„ZdS )é    N)ÚOrderedDict)ÚCallableÚTypeVar)Ú	ParamSpecÚsimple_call_counterÚ_PÚ_R)ÚlabelÚreturnc                 C   s   t  | d¡}|d t | < d S ©Nr   é   )r   Ú
setdefault)r	   Úprev© r   ú@/var/www/auris/lib/python3.9/site-packages/torch/utils/_stats.pyÚcount_label   s    r   )Úfnr
   c                    s(   t  ˆ ¡tjtjtdœ‡ fdd„ƒ}|S )N)ÚargsÚkwargsr
   c                     s6   ˆ j tvrdtˆ j < tˆ j  d tˆ j < ˆ | i |¤ŽS r   )Ú__qualname__r   )r   r   ©r   r   r   Úwrapper   s    

zcount.<locals>.wrapper)Ú	functoolsÚwrapsr   r   r   r   )r   r   r   r   r   Úcount   s    r   )Úcollectionsr   r   Útypingr   r   Ztyping_extensionsr   r   ÚstrÚintÚ__annotations__r   r   r   r   r   r   r   r   Ú<module>   s   
