a SicA@sVdZddlmmZddZddZddZdd d Zd d Z dd dZ ddZ dS)zjUtility functions for control flow. This file is copied from tensorflow/python/ops/control_flow_util.py. NcCs|}t|duSN)_get_control_flow_contextGetContainingXLAContext)graphctxtrY/var/www/html/django/DPS/env/lib/python3.9/site-packages/keras/utils/control_flow_util.py InXlaContextsr cCs2t|r dSz |j}Wqty*YdS0qdS)NTF)r outer_graphAttributeError)rrrrGraphOrParentsInXlaContexts   r cCs|}t|duSr)rGetContainingWhileContext)oprrrr IsInWhileLoop&srcCs$|r |s||kr|S|j}qdS)a1Returns 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_ctxtrrrr +s r cCs|r|r|S|j}qdS)aPReturns 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)rrrrrCs rcCs4t|tjrtj||||dStjjj||||dS)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) isinstancetfVariablecond __internal__ smart_cond)predrrrrrrrWs  rcCsZt|tjrt|S|dvr&t|St|tr4|St|tjrDdStdt|dS)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. >rNz@`pred` must be a Tensor, or a Python bool, or 1 or 0. Received: )rrTensorget_static_valueboolr TypeErrortype)rrrrconstant_valueqs    r$)N)NNN) __doc__tensorflow.compat.v2compatv2rr r rr rrr$rrrrs