o Me @sDddlmZdgZGdddeZeaddZGdddeZdS) ) framework LazyGuardc@sDeZdZdZddZddZddZdd Zd d Ze d d Z dS)LazyInitHelperz A Helper Context to trigger switching mode between dygraph and static mode, and holds the startup program resource. cCsd|_d|_d|_dS)NF)_state_tracer _in_guardselfr FD:\Projects\ConvertPro\env\Lib\site-packages\paddle/fluid/lazy_init.py__init__s zLazyInitHelper.__init__cCs$|jrdSts Jdd|_dS)zs Switch into lazy mode. NOTE(dev): This is a very low level API and not exposed for user. Nz4LazyInit.enable() is only available in dygraph mode.T)rrZ_non_static_moderr r r enables   zLazyInitHelper.enablecCs|jsdSd|_dS)zq Exit from lazy mode. NOTE(dev): This is a very low level API and not exposed for user. NFrrr r r disable+s zLazyInitHelper.disablecCs*||jr dStj|_dt_d|_dS)z| Switch into lazy mode and set _dygraph_tracer_ with None to convert dygraph mode into static mode. NT)r rr_dygraph_tracer_rrr r r __enter__5s   zLazyInitHelper.__enter__cOs8||js dS|jdusJ|jt_d|_d|_dS)zC Exit from lazy mode and recover _dygraph_tracer_. NF)rrrrrr argskwargsr r r __exit__@s   zLazyInitHelper.__exit__cCs|jSNrrr r r stateKszLazyInitHelper.stateN) __name__ __module__ __qualname____doc__r r rrrpropertyrr r r r rs    rcCstSr)_lazy_init_helperr r r r lazy_init_helperSsrc@s eZdZdZddZddZdS)rz LazyGuard is a wrapper interface for nn.Layer, it forwards the construct process of user defined Layer. Meanwhile, it provides necessary API to trigger EagerParamBase Lazy Initialization and get startup Program. cCstdS)a Construct instance from class_obj by Lazy Initializing parameters. Examples: .. code-block:: python from paddle import LazyGuard from paddle.nn import Linear with LazyGuard(): fc = LazyInit(Linear)(10, 10) for param in fc.parameters(): param.initialize() N)rr rr r r r_szLazyGuard.__enter__cOstdSr)rrrr r r rrszLazyGuard.__exit__N)rrrrrrr r r r rXs N)r__all__objectrrrrr r r r s <