o
    ´©iº  ã                   @   s    d dl Z G dd„ de jjƒZdS )é    Nc                   @   s$   e Zd Zedd„ ƒZedd„ ƒZdS )ÚGradMultiplyc                 C   s   || _ | |¡}|S ©N)ÚscaleÚnew)ÚctxÚxr   Úres© r	   úX/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/data2vec/grad_multiply.pyÚforward
   s   
zGradMultiply.forwardc                 C   s   || j  d fS r   )r   )r   Úgradr	   r	   r
   Úbackward   s   zGradMultiply.backwardN)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r	   r	   r	   r
   r   	   s
    
r   )ÚtorchÚautogradÚFunctionr   r	   r	   r	   r
   Ú<module>   s   