6biBLSrSSKJs Jr SrSrSrS SjrSr S Sjr S r g) zjUtility functions for control flow. This file is copied from tensorflow/python/ops/control_flow_util.py. Nc<UR5n[U5SL$N)_get_control_flow_contextGetContainingXLAContext)graphctxts ^/home/james-whalen/.local/lib/python3.13/site-packages/tf_keras/src/utils/control_flow_util.py InXlaContextr s  * * ,D "4 ( 44cd[U5(agURnM ![a gf=f)NTF)r outer_graphAttributeError)rs r GraphOrParentsInXlaContextrs<     %%E    s " //c<UR5n[U5SL$r)rGetContainingWhileContext)oprs r IsInWhileLoopr&s ' ' )D $T *$ 66r ctU(a1UR5(dX:XaU$URnU(aM1g)aReturns the first ancestor WhileContext of `ctxt`. Returns `ctxt` if `ctxt` is a WhileContext, or None if `ctxt` is not in a while loop. Args: ctxt: ControlFlowContext stop_ctxt: ControlFlowContext, optional. If provided, the search will end if it sees stop_ctxt. Returns: `ctxt` if `ctxt` is a WhileContext, the most nested WhileContext containing `ctxt`, or None if `ctxt` is not in a while loop. If `stop_ctxt` is not `None`, this returns `ctxt` if it matches `stop_ctxt` in its traversal. N)IsWhileContext outer_context)r stop_ctxts r rr+s7"     D$5K!! $ r cjU(a,UR5(aU$URnU(aM,g)a0Returns the first ancestor XLAContext of `ctxt`. Returns `ctxt` if `ctxt` is a XLAContext, or None if `ctxt` is not in a while loop. Args: ctxt: ControlFlowContext Returns: `ctxt` if `ctxt` is a XLAContext, the most nested XLAContext containing `ctxt`, or None if `ctxt` is not in a while loop. N) IsXLAContextr)rs r rrCs2      K!! $ r c[U[R5(a[R"XX#S9$[RR R XX#S9$)atReturn either `true_fn()` if predicate `pred` is true else `false_fn()`. If `pred` is a bool or has a constant value, we return either `true_fn()` or `false_fn()`, otherwise we use `tf.cond` to dynamically route to both. Args: pred: A scalar determining whether to return the result of `true_fn` or `false_fn`. true_fn: The callable to be performed if pred is true. false_fn: The callable to be performed if pred is false. name: Optional name prefix when using `tf.cond`. Returns: Tensors returned by the call to either `true_fn` or `false_fn`. Raises: TypeError: If `true_fn` or `false_fn` is not callable. )true_fnfalse_fnname) isinstancetfVariablecond __internal__ smart_cond)predrrrs r r#r#WsM&$ $$wwtxKK ?? % % 0 0  1 r c*[U[R5(a[R"U5$US;a [ U5$[U[5(aU$[U[R 5(ag[ S[U535e)axReturn the bool value for `pred`, or None if `pred` had a dynamic value. Args: pred: A scalar, either a Python bool or a TensorFlow boolean variable or tensor, or the Python integer 1 or 0. Returns: True or False if `pred` has a constant boolean value, None otherwise. Raises: TypeError: If `pred` is not a Variable, Tensor or bool, or Python integer 1 or 0. >rNz@`pred` must be a Tensor, or a Python bool, or 1 or 0. Received: )rrTensorget_static_valueboolr TypeErrortype)r$s r constant_valuer,qs}$ """"4(( v~Dz$ $ $$  $ZL " r r)NNN) __doc__tensorflow.compat.v2compatv2rr rrrrr#r,r r r2s5 "!5 7 0(4r