o "Õjñã@sZddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z Gdd„dƒZd S) é)Ú annotationsé)Úlogé)ÚCompileSIRCache)Ú ApiStatementÚ CallStatementÚLayerStatementÚMethodStatementÚ StatementIRÚStatementIRFactoryÚSymbolc@sveZdZdZdd„Zdd„Zedd„ƒZdd „Zd d „Z d d „Z dd„Z ddd„Z dd„Z dd„Zdd„Zdd„ZdS)ÚSymbolicTraceContextza SymbolicTraceContext is a context manager, which is used to record the symbolic trace. cCs | ¡dS©N)Úreset©Úself©rúi/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/sot/symbolic/symbolic_context.pyÚ__init__$s zSymbolicTraceContext.__init__cCstƒ|_|j ¡g|_dS)z$ Reset the context. N)r Ústatement_factoryÚcreateÚ sir_stackrrrrr'szSymbolicTraceContext.resetcCs |jdS)zh The top SIR of sir_stack. Returns: StatementIR: the top of stack. éÿÿÿÿ)rrrrrÚTOS1s zSymbolicTraceContext.TOScCót||||ƒ}|j |¡dS)z2 Call a SIR, which is a subgraph. N)rrÚ add_statement)rZsirnameÚinputsÚoutputsÚstacksÚstmtrrrÚcall_SIR<szSymbolicTraceContext.call_SIRcCs.t|ƒsJdƒ‚t||||ƒ}|j |¡dS)z$ Call a paddle api. z#call_API must receive a paddle api.N)Úcallablerrr)rÚapirrrr rrrÚcall_APIDszSymbolicTraceContext.call_APIcCsJt|tƒs Jdƒ‚t|ddtƒsJdƒ‚t||||ƒ}|j |¡dS)zN Call a method of a api. The API here can be python or Paddle z)call_METHOD must method api name. string.rz8call_METHOD must first augument must be Symbol Variable.N)Ú isinstanceÚstrr r rr)rÚ method_namerrrr rrrÚ call_METHODMsÿþ ÿþz SymbolicTraceContext.call_METHODcCr)z( Call a layer of a api. N)r rr)rÚlayerrrrr rrrÚ call_LAYERZszSymbolicTraceContext.call_LAYERÚnamer&cCs |j|S)zŸ Get a SIR from statement_factory. Args: name (str): the name of SIR. Returns: StatementIR: the SIR. )r)rr+rrrÚget_siras zSymbolicTraceContext.get_sircCs |j ¡|j |j ¡¡dS)z Reset the TOS. N)rÚpopÚappendrrrrrrÚ reset_TOSms zSymbolicTraceContext.reset_TOScCs&|j ¡|j |¡|j |¡dS)zl Use deepcopyed sir to replace the TOS. This function will update statment_factory. N)rr-r.rÚupdate)rZsirrrrÚ replace_TOSts  z SymbolicTraceContext.replace_TOScCs$dd„}tdƒ}g|_g|_||fS)z¢ Return a dummy function, which will return an empty list. Args: ret_vals (list[Symbol]): the return values of the function. c_sgSrr)ÚargsÚkwargsrrrÚ dummy_func…sz;SymbolicTraceContext.compile_do_nothing..dummy_funcr4)r rr)rÚret_valsr4Z dummy_stmt_irrrrÚcompile_do_nothing}s z'SymbolicTraceContext.compile_do_nothingcKsd|j}t|jƒdkr| |¡S| ¡|_||_tddƒtd|jdƒtƒ||j fi|¤Ž}||fS)zk start compile and return the python function, which must can be to_static without errors. rrz&start subgraph compile and execution. Ú ) rÚlenZ statementsr6Zanalyse_inputsrrrrr+)rr5r3Zcur_sirZ static_funcrrrÚ compile_fnŽs   zSymbolicTraceContext.compile_fnN)r+r&)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚpropertyrr!r$r(r*r,r/r1r6r9rrrrrs      rN)Ú __future__rÚutilsrZ compile_cacherZ statement_irrrr r r r r rrrrrÚs   $