BVhdZddlmZddlZddlmZmZmZddlm Z ddl m Z edZ Gdd ejee ZeZd Ze d g d Zy)z+Classes used to handle thread-local stacks.)IteratorN)GenericOptionalTypeVar) tf_contextlib) tf_exportTceZdZdZfdZdeefdZd dZde fdZ e de fdZ e jde fd Z ejd edeefd ZxZS) DefaultStackz@A thread-local stack of objects for providing implicit defaults.c>t|d|_g|_y)NT)super__init___enforce_nestingstack)self __class__s Q/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/stack.pyrzDefaultStack.__init__s G DDJreturnc<|jr|jdSdS)Nrrs r get_defaultzDefaultStack.get_default#s!ZZ4::b>1T1rcg|_yNrrs rresetzDefaultStack.reset&s DJrc|j Srrrs r is_clearedzDefaultStack.is_cleared)szz>rc|jSrrrs renforce_nestingzDefaultStack.enforce_nesting,s   rvaluec||_yrr!)rr#s rr"zDefaultStack.enforce_nesting0s !Drdefaultc#2K|jj| ||jrk|jrC|jd|urtdt |z|jj y|jj |yy#|jrk|jrC|jd|urtdt |z|jj w|jj |wwxYww)z3A context manager for manipulating a default stack.rz0Nesting violated for default stack of %s objectsN)rappendrAssertionErrortypepopremove)rr%s rget_controllerzDefaultStack.get_controller4s JJg % m   ZZ^7 * BW   **..  **  G $   ZZ^7 * BW   **..  **  G $ sDBA8DA9DD)rN)__name__ __module__ __qualname____doc__rrr rrboolrpropertyr"setterrcontextmanagerrr, __classcell__)rs@rr r sH 28A;2$ !t! !"4""%A%(1+% %rr c,tj|S)aPython "with" handler for defining a default session. This function provides a means of registering a session for handling Tensor.eval() and Operation.run() calls. It is primarily intended for use by session.Session, but can be used with any object that implements the Session.run() interface. Use with the "with" keyword to specify that Tensor.eval() and Operation.run() invocations within the scope of a block should be executed by a particular session. The default session applies to the current thread only, so it is always possible to inspect the call stack and determine the scope of a default session. If you create a new thread, and wish to use the default session in that thread, you must explicitly add a "with ops.default_session(sess):" block in that thread's function. Example: The following code examples are equivalent: # 1. Using the Session object directly: sess = ... c = tf.constant(5.0) sess.run(c) # 2. Using default_session(): sess = ... with ops.default_session(sess): c = tf.constant(5.0) result = c.eval() # 3. Overriding default_session(): sess = ... with ops.default_session(sess): c = tf.constant(5.0) with ops.default_session(...): c.eval(session=sess) Args: session: The session to be installed as the default session. Returns: A context manager for the default session. )_default_session_stackr,)sessions rdefault_sessionr9JsZ . .w 77rget_default_session)v1c*tjS)aReturns the default session for the current thread. The returned `Session` will be the innermost session on which a `Session` or `Session.as_default()` context has been entered. NOTE: The default session is a property of the current thread. If you create a new thread, and wish to use the default session in that thread, you must explicitly add a `with sess.as_default():` in that thread's function. Returns: The default `Session` being used in the current thread. )r7rrrr:r:zs + + --r)r0collections.abcr threadingtypingrrrtensorflow.python.utilr tensorflow.python.util.tf_exportrr localr r7r9r:r=rrrDsi2$--06 CL)%9??GAJ)%X&-8` $%&.'.r