o
    "Õjz  ã                   @   s<   d dl mZ d dlmZ d dlmZ g ZG dd„ deƒZdS )é   )ÚBaseTransformer)ÚFunctionNameLivenessAnalysis)Úcreate_undefined_varc                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	ÚCreateVariableTransformerú c                 C   s   || _ t| j ƒ d S )N)Úrootr   )Úselfr   © r	   úq/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/dy2static/create_variable_transformer.pyÚ__init__   s   z"CreateVariableTransformer.__init__c                 C   s   |   | j¡ dS )z1
        Main function to transform AST.
        N)Úvisitr   )r   r	   r	   r
   Ú	transform   s   z#CreateVariableTransformer.transformc                 C   s>   |   |¡ |j}t|j ¡ ƒ}|D ]}t|ƒg|dd…< q|S )Né    )Úgeneric_visitÚbodyÚsortedZpd_scopeZcreated_varsr   )r   ÚnodeZbodysÚnamesÚnamer	   r	   r
   Úvisit_FunctionDef#   s   
z+CreateVariableTransformer.visit_FunctionDefN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r	   r	   r	   r
   r      s
    r   N)Zbase_transformerr   Úutilsr   Zvariable_trans_funcr   Ú__all__r   r	   r	   r	   r
   Ú<module>   s
   