o ?e@sldZddlmZddlmZddlmZddlmZddlmZedgddd d Z d d Z dddZ d S)z!smart_cond and related utilities.)tensor) tensor_util)cond)control_flow_case) tf_exportz"__internal__.smart_cond.smart_cond)v1NcCs^t|s td|t|std|t|}|dur&|r#|S|Stj||||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. z.Argument `true_fn` must be callable. Received z/Argument `false_fn` must be callable. Received N)true_fnfalse_fnname)callable TypeErrorsmart_constant_valuer)predrr r pred_valuerg/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/smart_cond.py smart_conds rcCslt|tjrt|}|durt|}|S|dvr t|}|St|tr)|}|Std|dt|j )aReturn the bool value for `pred`, or None if `pred` had a dynamic value. Args: pred: A scalar, either a Python bool or tensor. Returns: True or False if `pred` has a constant boolean value, None otherwise. Raises: TypeError: If `pred` is not a Tensor or bool. N>rzNArgument `pred` must be a Tensor, or a Python bool, or 1 or 0. Received: pred=z of type ) isinstancerZTensorrZconstant_valueZtry_evaluate_constantboolr type__name__)rrrrrr =s    r F smart_casecCstjt||||ddS)aLike tf.case, except attempts to statically evaluate predicates. If any predicate in `pred_fn_pairs` is a bool or has a constant value, the associated callable will be called or omitted depending on its value. Otherwise this functions like tf.case. Args: pred_fn_pairs: Dict or list of pairs of a boolean scalar tensor and a callable which returns a list of tensors. default: Optional callable that returns a list of tensors. exclusive: True iff at most one predicate is allowed to evaluate to `True`. name: A name for this operation (optional). Returns: The tensors returned by the first pair whose predicate evaluated to True, or those returned by `default` if none does. Raises: TypeError: If `pred_fn_pairs` is not a list/dictionary. TypeError: If `pred_fn_pairs` is a list but does not contain 2-tuples. TypeError: If `fns[i]` is not callable for any i, or `default` is not callable. T)Zallow_python_preds)rZ _case_helperr)Z pred_fn_pairsdefaultZ exclusiver rrrr\s)NNN)NFr) __doc__Ztensorflow.python.frameworkrrZtensorflow.python.opsrrZ tensorflow.python.util.tf_exportrrr rrrrrs       $