
    `i                     >    d dl Z  G d de j        j                  ZdS )    Nc                   $     e Zd Z fdZd Z xZS )_WrapperModulec                 V    t                                                       || _        d S N)super__init__f)selfr	   	__class__s     o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/export/_wrapper_utils.pyr   z_WrapperModule.__init__   s$        c                      | j         |i |S r   )r	   )r
   argskwargss      r   forwardz_WrapperModule.forward	   s    tvt&v&&&r   )__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      sG            ' ' ' ' ' ' 'r   r   )torchnnModuler    r   r   <module>r      sF    ' ' ' ' 'UX_ ' ' ' ' 'r   