BVhtdZddlmZddlmZddlmZddlmZddlmZedgd d Z d Z dd Z y )z!smart_cond and related utilities.)tensor) tensor_util)cond)control_flow_case) tf_exportz"__internal__.smart_cond.smart_cond)v1Nct|std|t|std|t|}||r|S|Stj||||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. z.Argument `true_fn` must be callable. Received z/Argument `false_fn` must be callable. Received )true_fnfalse_fnname)callable TypeErrorsmart_constant_valuer)predr r r pred_values V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/smart_cond.py smart_condrs( '  DWIN OO (   9(D FF$D)* Y Z 99T7X  c$t|tjr.tj|}|tj |}|S|dvr t |}|St|t r|}|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. >rzNArgument `pred` must be a Tensor, or a Python bool, or 1 or 0. Received: pred=z of type ) isinstancerTensorrconstant_valuetry_evaluate_constantboolrtype__name__)rrs rrr=sfmm$++D1J44T:j  v~dJ $J  ,,06Dz**+- ..rc@tjt||||dS)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)allow_python_preds)r _case_helperr) pred_fn_pairsdefault exclusiver s r smart_caser$\s)0  ' '    r)NNN)NFr$) __doc__tensorflow.python.frameworkrrtensorflow.python.opsrr tensorflow.python.util.tf_exportrrrr$rrr*s@(.3&36 /B7! 8! H>r