BVhrdZddlmZddlmZddlmZddlmZddlmZdZ dZ d Z dd Z d Z dd ZdZy )zjUtility functions for control flow. This file is copied from tensorflow/python/ops/control_flow_util.py. ) smart_cond)tensor) tensor_util)cond) variablesc<|j}t|duSN)_get_control_flow_contextGetContainingXLAContext)graphctxts _/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/keras/utils/control_flow_util.py InXlaContextrs  ( ( *$  &d 22cV t|ry |j}#t$rYywxYw)NTF)r outer_graphAttributeError)r s rGraphOrParentsInXlaContextr s8 E4e   s  ((c<|j}t|duSr )r GetContainingWhileContext)opr s r IsInWhileLoopr)s % % '$ "4 ( 44rNcT|r&|js||k(r|S|j}|r&y)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 rrr.s3   1$;   D  rcJ|r!|jr|S|j}|r!y)a@Returns 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)r s rr r Ds,  4K   D  rct|tjrtj||||St j ||||S)aReturn 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) isinstancerVariabler smart_moduler)predr r!r"s rrrWsJ&i(() 99 gt ==   GhT ;;rct|tjrtj|S|dvr t |St|t r|St|t jrytdt|z)aReturn 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. >rNzG`pred` must be a Tensor, or a Python bool, or 1 or 0. Found instead: %s) r#rTensorrconstant_valueboolrr$ TypeErrortype)r&s rr*r*qstfmm$  % %d ++ V^ :d Ki(() &(,T 3 44rr )NNN)__doc__tensorflow.python.frameworkrr%rrtensorflow.python.opsrrrrrrr r*rrr2s? C.3&+3 5 ,&;44r