AVh1dZddlZddlZddlmZddlmZejxrejddk7xs^ejdddk7xsFejdddk7xs.ejdddk7xsejd ddk7a d Z d Z d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZd$dZdZdZdZ d Z!d!Z"d"Z#d#Z$y)%zUtility functions for control flow. This file is necessary to avoid cyclic dependencies between ops.py and control_flow_ops.py. N)tf2) tf_loggingTF_ENABLE_CONTROL_FLOW_V20TF_ENABLE_COND_V2TF_ENABLE_WHILE_V2TF_ENABLE_TENSOR_ARRAY_V2cday)zHUse control flow v2. Do not use this symbol. This will be removed. TN)ENABLE_CONTROL_FLOW_V2W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/control_flow_util.pyenable_control_flow_v2r%s  r cHtxs|jxr t|d S)z:Returns whether control flow v2 should be used in `graph`. _captured)r building_functionhasattrgraphs rEnableControlFlowV2r.s+  C Age[&A"ACr c |jd}|ry |j}t|duS#t$rY(wxYw)N _XlaCompileT)get_attr ValueError_get_control_flow_contextGetContainingXLAContext)op xla_compilectxts rIsInXLAContextr 6sP ++m,K4{ % % '$  &d 22   s4 AAc<|j}t|duSN)rr)rrs r InXlaContextr#@s  ( ( *$  &d 22r cV t|ry |j}#t$rYywxYw)NTF)r# outer_graphAttributeErrorrs rGraphOrParentsInXlaContextr'Es8 E4e   s  ((c<|j}t|duSr")rGetContainingWhileContextrrs r IsInWhileLoopr+Ns % % '$ "4 ( 44r c<|j}t|duSr")rGetContainingCondContextr*s rIsInCondr.Ss % % '$ !$ 't 33r cB|jdk(xs|jdk(S)z Return true if `op` is a Switch.Switch RefSwitchtypers rIsSwitchr5Xs H  6; 66r cB|jdk(xs|jdk(S)zReturn true if `op` is a Merge.MergeRefMerger2r4s rIsMerger9] G  4rww*44r cB|jdk(xs|jdk(S)z!Returns true if `op` is an Enter.EnterRefEnterr2r4s r IsLoopEnterr>br:r cB|jdk(xs|jdk(S)zReturn true if `op` is an Exit.ExitRefExitr2r4s r IsLoopExitrBgs F  2bgg22r c t|sy|jsyd}|jD]X}|jD]C}|j}t |r |j }|xr|duxr|j }EZ|S)z4Return true if `op` is the Switch for a conditional.FTN)r5outputs consumersrr> outer_context IsCondContext)ris_cond_switchocrs r IsCondSwitchrKls "    . ::Aa [[]A  ( ( *d Q!!%A4t+;,@+/+=+=+? AA r ct|sy|jsyd}|jD]1}t|j}|xr|duxr|j }3|S)z3Return true if `op` is the Merge for a conditional.FTN)r9inputsGetOutputContextrrG)r is_cond_mergeirs r IsCondMergerQse     - 99Pa ADD !D!Od$&6O4;M;M;OMP r ct|r4|j}|duxr|jxr t| Sy)z3Return true if `op` is the Switch for a while loop.NF)r5rIsWhileContextrKr*s r IsLoopSwitchrTs@ b\ ' ' )D t  N 3 3 5 Nl2>N:NN r ct|r4|j}|duxr|jxr t| Sy)z2Return true if `op` is the Merge for a while loop.NF)r9rrSrQr*s r IsLoopMergerVs? R[ ' ' )D t  M 3 3 5 Mk"o:MM r c>t|xr|jdS)z'Return true iff op is a loop invariant. is_constant)r>rr4s rIsLoopConstantEnterrYs R 7R[[77r chd}|j}|j|vr(|jdj}|j|vr(t|r|SdS)zCReturn the enter op if we can infer `value` to be a loop invariant.>r0Identityr1 RefIdentityrN)rr3rMrY)valueid_opsrs rGetLoopConstantEnterr_sM =& xx" 6 1B 6"2&0D0r cX|j}|t|r |j}|S)z8Return the control flow context for the output of an op.)rrBrFr*s rrNrNs. % % '$ *R.   D +r cT|r&|js||k(r|S|j}|r&y)aReturns 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)rSrF)r stop_ctxts rr)r)s3   1$;   D  r cJ|r!|jr|S|j}|r!y)a@Returns 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) IsXLAContextrFrs rrrs,  4K   D  r cJ|r!|jr|S|j}|r!y)a6Returns the first ancestor CondContext of `ctxt`. Returns `ctxt` if `ctxt` is a CondContext, or None if `ctxt` is not in a cond. Args: ctxt: ControlFlowContext Returns: `ctxt` if `ctxt` is a CondContext, the most nested CondContext containing `ctxt`, or None if `ctxt` is not in a cond. N)rGrFres rr-r-s,  D[   D  r c4||ur|y|j}||ury)z>Returns true if `maybe_containing_ctxt` is or contains `ctxt`.FT)rF)rmaybe_containing_ctxts rIsContainingContextris- )) |E   D )) r c6t|j|Sr")rirr*s r OpInContextrks R99;T BBr c.t|j|Sr")rkr)tensorrs rTensorInContextrns VYY %%r c |j}t|}d}|sd}n/||urd}n't|}t|}||d}t|rd}t |rd}nt ||rd}n|j r#t |j j|rd}n|j r%|j j|jurd}n|j r|j j|urd}nZ|j rN|j jj r.|j jj j|urd}|srd|jd|jd}n*d|jd|jd|jd}|}|d |jd |z }|d |jd z }|d |jd djtj|jd|jd djtj|jd z }tj|t|dzy)anReturns whether `input_op` can be used from `op`s context. Conceptually, only inputs from op's while context or any ancestor while context (including outside of any context) are valid. In practice, there are many other edge cases as well. Args: op: Operation input_op: Operation Raises: ValueError: if input_op is from an invalid context. FTNz Cannot use 'z' as input to 'z,' because they are in different while loops.z ' because 'z' is in a while loop.z z while context:  z Traceback for z: z Traceback for z See info log for more details.)rrNr)r>r5ri grad_stateforward_context_outer_contextnamejoin traceback format_listlogginginfor) rinput_opop_ctxt input_ctxtvalid while_ctxtinput_while_ctxt error_msglog_msgs rCheckInputFromValidContextrsO  ( ( *')* %  E* E*73J0<  ! R " Z)9 :e    j33CC. 0e       / /  ) ) * e  % %  % % 5 5 Ce  % %    / / : :    / / : : J J   e  rwwi@* * rwwi@mm_1 3G RWWjAAG 8==:JKKG  ..r||<= rwwy44X5G5GHIKKG LL Y!BB CC) r cJ|j}|r|j}|S)z.Get the WhileContext to which this op belongs.)rGetWhileContextr*s rrrjs& % % '$    !D +r r")%__doc__osrwtensorflow.pythonrtensorflow.python.platformrryenabledgetenvr rrr r#r'r+r.r5r9r>rBrKrQrTrVrYr_rNr)rr-rirkrnrrr r rrs?   !<&3;;=I$299%@ASHM#"))$?ELM$"))$7=DM$"))$8#>#E M $"))$?EL  C33 5 4 7 5 5 3 *"8 1,&$C&aDHr