AVh;TdZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l m Z dd l mZdd lmZdd lmZddlmZedgej, ddZedgej, ddZdZdZ ddZdZy)z+Case functions for Control Flow Operations.N)context) constant_op)dtypes)ops)tensor)array_ops_stack)cond)control_flow_assert)math_ops) tf_logging)dispatch) tf_exportcase)v1c Bttj||||d|S)ap Create a case operation. See also `tf.switch_case`. The `pred_fn_pairs` parameter is a list of pairs of size N. Each pair contains a boolean scalar tensor and a python callable that creates the tensors to be returned if the boolean evaluates to True. `default` is a callable generating a list of tensors. All the callables in `pred_fn_pairs` as well as `default` (if provided) should return the same number and types of tensors. If `exclusive==True`, all predicates are evaluated, and an exception is thrown if more than one of the predicates evaluates to `True`. If `exclusive==False`, execution stops at the first predicate which evaluates to True, and the tensors generated by the corresponding function are returned immediately. If none of the predicates evaluate to True, this operation returns the tensors generated by `default`. `tf.case` supports nested structures as implemented in `tf.nest`. All of the callables must return the same (possibly nested) value structure of lists, tuples, and/or named tuples. Singleton lists and tuples form the only exceptions to this: when returned by a callable, they are implicitly unpacked to single values. This behavior is disabled by passing `strict=True`. @compatibility(v2) `pred_fn_pairs` could be a dictionary in v1. However, tf.Tensor and tf.Variable are no longer hashable in v2, so cannot be used as a key for a dictionary. Please use a list or a tuple instead. @end_compatibility **Example 1:** Pseudocode: ``` if (x < y) return 17; else return 23; ``` Expressions: ```python f1 = lambda: tf.constant(17) f2 = lambda: tf.constant(23) r = tf.case([(tf.less(x, y), f1)], default=f2) ``` **Example 2:** Pseudocode: ``` if (x < y && x > z) raise OpError("Only one predicate may evaluate to True"); if (x < y) return 17; else if (x > z) return 23; else return -1; ``` Expressions: ```python def f1(): return tf.constant(17) def f2(): return tf.constant(23) def f3(): return tf.constant(-1) r = tf.case([(tf.less(x, y), f1), (tf.greater(x, z), f2)], default=f3, exclusive=True) ``` Args: pred_fn_pairs: 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`. strict: A boolean that enables/disables 'strict' mode; see above. 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/tuple. 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. Fallow_python_predsstrict _case_helperr  pred_fn_pairsdefault exclusivernames W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/control_flow_case.pycase_v2r!s+~  ii    c Bttj||||d|S)a4 Create a case operation. See also `tf.switch_case`. The `pred_fn_pairs` parameter is a dict or list of pairs of size N. Each pair contains a boolean scalar tensor and a python callable that creates the tensors to be returned if the boolean evaluates to True. `default` is a callable generating a list of tensors. All the callables in `pred_fn_pairs` as well as `default` (if provided) should return the same number and types of tensors. If `exclusive==True`, all predicates are evaluated, and an exception is thrown if more than one of the predicates evaluates to `True`. If `exclusive==False`, execution stops at the first predicate which evaluates to True, and the tensors generated by the corresponding function are returned immediately. If none of the predicates evaluate to True, this operation returns the tensors generated by `default`. `tf.case` supports nested structures as implemented in `tf.nest`. All of the callables must return the same (possibly nested) value structure of lists, tuples, and/or named tuples. Singleton lists and tuples form the only exceptions to this: when returned by a callable, they are implicitly unpacked to single values. This behavior is disabled by passing `strict=True`. If an unordered dictionary is used for `pred_fn_pairs`, the order of the conditional tests is not guaranteed. However, the order is guaranteed to be deterministic, so that variables created in conditional branches are created in fixed order across runs. @compatibility(eager) Unordered dictionaries are not supported in eager mode when `exclusive=False`. Use a list of tuples instead. @end_compatibility **Example 1:** Pseudocode: ``` if (x < y) return 17; else return 23; ``` Expressions: ```python f1 = lambda: tf.constant(17) f2 = lambda: tf.constant(23) r = tf.case([(tf.less(x, y), f1)], default=f2) ``` **Example 2:** Pseudocode: ``` if (x < y && x > z) raise OpError("Only one predicate may evaluate to True"); if (x < y) return 17; else if (x > z) return 23; else return -1; ``` Expressions: ```python def f1(): return tf.constant(17) def f2(): return tf.constant(23) def f3(): return tf.constant(-1) r = tf.case({tf.less(x, y): f1, tf.greater(x, z): f2}, default=f3, exclusive=True) ``` 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`. strict: A boolean that enables/disables 'strict' mode; see above. 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. Frrrs rrrs+F  ii    rc~tj|d}tjtj|t j d}tj|tj|d}djd|D}d|||fz|g}tj||t|S) zReturns an Assert op that checks that at most n predicates are True. Args: predicates: list of bool scalar tensors. n: maximum number of true predicates allowed. msg: Error message. preds_crnum_true_conds n_true_condsz, c36K|]}t|ddyw)r?N)getattr).0ps r z)_assert_at_most_n_true..sFa'!VS1Fsz3%s: more than %d conditions (%s) evaluated as True:)data summarize)rstackr reduce_sumcastrint32 less_equalrconstantjoinr Assertlen) predicatesnmsgr!num_true_conditions condition preds_names error_msgs r_assert_at_most_n_truer=s  ! !*9 =' ++mmGV\\*1AC!!"5"-"6"6q~"NP) F:FF+; A{$)  # #i3z? <.default_actions*,-.J;"(()DFK ! ! z 3""9;?#  X s A))A2)r5)r6actions other_actionsrIrErFrGrHs` @@@@r_case_create_default_actionrL sZ *o! mWQZ)V$.rNGBQK!M   )= 88rc t||||\}}tj|d|g5|t||\}}}|} t t t ||D]\} } tj|| f| | d|} !|r. z:'pred_fn_pairs' must be a list, tuple, or dict. Received: z{Unordered dictionaries are not supported for the 'pred_fn_pairs' argument when `exclusive=False` and eager mode is enabled.c |djS)Nrr")items rz4_case_verify_and_canonicalize_args..s$q',,r)keyz%s: An unordered dictionary of predicate/fn pairs was provided, but exclusive=False. The order of conditional tests is deterministic but not guaranteed.z:Each entry in 'pred_fn_pairs' must be a 2-tuple. Received .z$pred must be Tensor of type bool: %szpred must be a Tensor, got: %sz&pred must be a Tensor or bool, got: %sz fn for pred %s must be callable.) isinstancerStupledict TypeErrortype collections OrderedDictitemsrexecuting_eagerly ValueErrorsortedloggingwarnr5rTensordtyperboolrcallablerT) rrrr pred_fn_pairpredrYr6rJs rrPrPcs* MD%#6 7 !!%m!4 57 88 {667!'')M-&  "23 3=..01m    %>@m   9:> @$Fl lE *c,.?1.D "".q2 33HD"$ & v{{ ">JKK  6= >> d # >E FF B< 8499D EEF"]+*g W r)NFFr)F)__doc__rfrUtensorflow.python.eagerrtensorflow.python.frameworkrrrrtensorflow.python.opsrr r r tensorflow.python.platformr rltensorflow.python.utilr tensorflow.python.util.tf_exportradd_dispatch_supportrrr=rLrrPrrr}s2+3.+.1&5*<+6 6b  ddN vh  hhV<,9N%* 1h>r