
    zj+                     l    d dl mZ d dlmZ  G d d          Z G d d          Z G d d          Zd	S )
    )core)Loadc                       e Zd Zd Zd ZdS )Layerc                 "    d | _         i | _        d S N)	cpp_layer	functionsselfs    `/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/jit/layer.py__init__zLayer.__init__   s        c                 8   t          ||          | _        | j                                        D ]j}| j                            |          }| j                            |          }t          ||          | j        |<   t          | || j        |                    kd S r   )r   r	   function_namesfunctionfunction_infoFunctionr
   setattr)r   	load_pathplacenamer   infos         r   loadz
Layer.load   s    i//N1133 	6 	6D~..t44H>//55D#+Hd#;#;DN4 D$t 45555		6 	6r   N)__name__
__module____qualname__r   r    r   r   r   r      s2          
6 6 6 6 6r   r   c                       e Zd Zd Zd ZdS )r   c                 <    || _         t          |          | _        d S r   )r   FunctionInfor   )r   r   r   s      r   r   zFunction.__init__%   s      &&			r   c                 L    t           j                            | j        |          S r   )r   eagerjit_function_callr   )r   argss     r   __call__zFunction.__call__)   s    z++DM4@@@r   N)r   r   r   r   r&   r   r   r   r   r   $   s7        ' ' 'A A A A Ar   r   c                       e Zd Zd Zd ZdS )r!   c                     || _         d S r   )r   )r   r   s     r   r   zFunctionInfo.__init__.   s    			r   c                 4    | j                                         S r   )r   r   r   s    r   r   zFunctionInfo.name1   s    y~~r   N)r   r   r   r   r   r   r   r   r!   r!   -   s2                   r   r!   N)paddle.baser   paddle.base.corer   r   r   r!   r   r   r   <module>r,      s           ! ! ! ! ! !6 6 6 6 6 6 6 6 A A A A A A A A                   r   