
    BVh%                     >    d Z ddlmZ ddlmZ ddlmZ d Zd Z	d Z
y)	z8Conditional expressions (e.g. the ternary if statement).    )control_flow)tensors)condc                 b    t        j                  |       rt        | |||      S t        | ||      S N)r   is_dense_tensor
_tf_if_exp
_py_if_exp)r   if_trueif_false	expr_reprs       m/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/operators/conditional_expressions.pyif_expr      s0    T"dGXy99dGX..    c                 b    g g fd}fd}t        j                  | ||      S )z)Overload of if_exp that stages a TF cond.c                  |    j                                 rrt        j                   d   d          d   S Nr   appendr   verify_single_cond_var)r   	false_valr   true_vals   r   true_fnz_tf_if_exp.<locals>.true_fn$   s:    OOGII)))Xa[)A,OA;r   c                  |    j                                 rrt        j                   d   d          d   S r   r   )r   r   r   r   s   r   false_fnz_tf_if_exp.<locals>.false_fn*   s<    XZ I)))Xa[)A,OQ<r   )tf_condr   )r   r   r   r   r   r   r   r   s    ```  @@r   r	   r	      s.     () 
dGX	..r   c                 "    | r |       S  |       S r    )r   r   r   s      r   r
   r
   3   s    *
*r   N)__doc__%tensorflow.python.autograph.operatorsr   !tensorflow.python.autograph.utilsr   tensorflow.python.opsr   r   r   r	   r
   r   r   r   <module>r#      s$    ? ? 5 1//*+r   