{-jddlmZddlmZddlmZerddlmZdgZGddZ e a d Z Gd dZ d S) ) annotations) TYPE_CHECKING) framework) TracebackType LazyGuardcFeZdZdZdZdZdZdZdZe dZ dS) LazyInitHelperz A Helper Context to trigger switching mode between dygraph and static graph mode, and holds the startup program resource. c0d|_d|_d|_dS)NF)_state_tracer _in_guardselfs _/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/nn/initializer/lazy_init.py__init__zLazyInitHelper.__init__"s  c`|jrdStjs 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)r rin_dygraph_moders renablezLazyInitHelper.enable'sF ;  F(**  B  * rc&|jsdSd|_dS)zq Exit from lazy mode. NOTE(dev): This is a very low level API and not exposed for user. NFr rs rdisablezLazyInitHelper.disable4s {  F rc||jrdStjj|_dtj_d|_dS)z Switch into lazy mode and set _dygraph_tracer_ with None to convert dygraph mode into static graph mode. NT)rrr global_var_dygraph_tracer_r rs r __enter__zLazyInitHelper.__enter__>sC >  F +< 04 -rc||jsdS|jJ|jtj_d|_d|_dS)zC Exit from lazy mode and recover _dygraph_tracer_. NF)rrr rrr)rargskwargss r__exit__zLazyInitHelper.__exit__JsM ~  F|'''04 - rc|jSNrrs rstatezLazyInitHelper.stateVs {rN) __name__ __module__ __qualname____doc__rrrrr!propertyr$rrr r s          Xrr ctSr#)_lazy_init_helperr*rrlazy_init_helperr-^s rc"eZdZdZd dZdd Zd S)ra 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. Examples: .. code-block:: python >>> from paddle import LazyGuard >>> from paddle.nn import Linear >>> with LazyGuard(): ... # w and b are initialized lazily and have no memory. ... net = Linear(10, 10) ... >>> for param in net.parameters(): ... # Initialize param and allocate memory explicitly. ... param.initialize() returnNonecFtdS)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)r-rrs rrzLazyGuard.__enter__ys"" !!#####rexc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NonecFtdSr#)r-r)rr2r4r6s rr!zLazyGuard.__exit__s" ""$$$$$rN)r/r0)r2r3r4r5r6r7r/r0)r%r&r'r(rr!r*rrrrcsF*$$$$&%%%%%%rN) __future__rtypingrbasertypesr__all__r r,r-rr*rrr>s#""""" $###### -<<<<<<<<~#N$$ /%/%/%/%/%/%/%/%/%/%r