AVhX7dZddlmZddlmZddlmZdZdZdZ dZ dd Z d Z dd Z d ZdZddZdZ ddZGddeZdZdZdZdZy )z#Tools for selecting ops in a graph.)ops)tensor)object_identityc tjj|jjduS#t $rYywxYwNF)r_gradient_registrylookupop_defname LookupErrorops Q/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/op_selector.pyis_differentiablers=  ! ! ( ( 8 DD  s47 AAcpt|tjry t|}y#t$rYywxYw)z&Return true if the object is iterable.FT) isinstance tensor_libTensoriter Exception)obj_s r is_iterablers<Z&&'  S A   s ) 55cxt|}|D])}||vs|j||j|+|S)aAdd all the elements of `lb` to `la` if they are not there already. The elements added to `la` maintain ordering with respect to `lb`. Args: la: List of Python objects. lb: List of Python objects. Returns: `la`: The list `la` with missing elements from `lb`. )setappendadd)lalbla_setls rconcatenate_uniquer"(s@ r7& aiil jjm )ct|tjs#tdj t |g}|j D]}||jz }|S)zget all the tensors which are input or output of an op in the graph. Args: graph: a `tf.Graph`. Returns: A list of `tf.Tensor`. Raises: TypeError: if graph is not a `tf.Graph`. zExpected a graph, got: {})rrGraph TypeErrorformattypeget_operationsoutputs)graphtsrs r get_tensorsr-;s] E399 % /66tE{C DD "  "b"**B )r#Nc zt|tjr|St|s#t dj t ||!tjtjf}nt|s|f}d}|D]}t||sJt dj dj|Dcgc] }t|c}t || |j}h|j|jjk7stdj || |s td|Scc}w)a<Return the unique graph used by the all the elements in tops. Args: tops: iterable of elements to check (usually a list of tf.Operation and/or tf.Tensor). Or a tf.Graph. check_types: check that the element in tops are of given type(s). If None, the types (tf.Operation, tf.Tensor) are used. none_if_empty: don't raise an error if tops is an empty list, just return None. Returns: The unique graph used by all the tops. Raises: TypeError: if tops is not a iterable of tf.Operation. ValueError: if the graph is not unique. z{} is not iterableNz Expected a type in ({}), got: {}z, z+Operation {} does not belong to given graphz,Can't find the unique graph of an empty list)rrr%rr&r'r( Operationrrjoinstrr+ _graph_key ValueError)tops check_types none_if_emptygrts rget_unique_graphr9Ms/ cii K T  (//T ; <<==*"3"34K { #.K ! Qb b+ & 8?? !K#LO L K#A$%)"X/ 00y ((a ,, , DKKBO PPQY} C DD (K#s.D8cd}t|D]J\}}||j |j}!|j.|j|us=td|dy)zCheck that all the element in args belong to the same graph. Args: *args: a list of object with a obj.graph property. Raises: ValueError: if all the elements do not belong to the same graph. Nzargs[z7] does not belong to the same graph as other arguments.) enumerater+r3)argsr+isgvs r check_graphsr?ssi %$+fa }.iie  399E#9 qc"** ++ +r#c0t|tjr|r t|St dt |s|g}|sgS|r!|rdnt j}t|||Dcgc]}t|t js|!c}Scc}w)aConvert ts to a list of `tf.Tensor`. Args: ts: can be an iterable of `tf.Tensor`, a `tf.Graph` or a single tensor. check_graph: if `True` check if all the tensors belong to the same graph. allow_graph: if `False` a `tf.Graph` cannot be converted. ignore_ops: if `True`, silently ignore `tf.Operation`. Returns: A newly created list of `tf.Tensor`. Raises: TypeError: if `ts` cannot be converted to a list of `tf.Tensor` or, if `check_graph` is `True`, if all the ops do not belong to the same graph. 0allow_graph is False: cannot convert a tf.Graph.Nr5) rrr%r-r&rrrr9)r, check_graph allow_graph ignore_opsr5r8s rmake_list_of_trFsCII _ H II r? 4b i&DJ,=,=kr{3 >!Z:+<+<=A >> >s ,B BcZt|d}|Dcgc]}|jc}Scc}w)aReturn all the generating ops of the tensors in `ts`. Args: ts: a list of `tf.Tensor` Returns: A list of all the generating `tf.Operation` of the tensors in `ts`. Raises: TypeError: if `ts` cannot be converted to a list of `tf.Tensor`. FrD)rFr)r,r8s rget_generating_opsrIs(be," 1!$$  s(ct|d}g}|D]-}|jD]}||vs|j|/|S)zReturn all the consuming ops of the tensors in ts. Args: ts: a list of `tf.Tensor` Returns: A list of all the consuming `tf.Operation` of the tensors in `ts`. Raises: TypeError: if ts cannot be converted to a list of `tf.Tensor`. FrH)rF consumersr)r,r4r8rs rget_consuming_opsrLsRbe," $ akkm 4 B +r#c:t|tjr|r|jSt dt |s|g}|sgS|r!|rdntj }t|||Dcgc]}t|tj s|!c}Scc}w)a4Convert ops to a list of `tf.Operation`. Args: tops: can be an iterable of `tf.Operation`, a `tf.Graph` or a single operation. check_graph: if `True` check if all the operations belong to the same graph. allow_graph: if `False` a `tf.Graph` cannot be converted. ignore_ts: if True, silently ignore `tf.Tensor`. Returns: A newly created list of `tf.Operation`. Raises: TypeError: if tops cannot be converted to a list of `tf.Operation` or, if `check_graph` is `True`, if all the ops do not belong to the same graph. rANrB)rrr%r)r&rr/r9)r4rCrD ignore_tsr5rs rmake_list_of_oprOs cii   "" H II t Vd  i%D3==kt5 ?2B !>B ?? ?s 1BBc@|j}|rt|r|SgS|SN)inputsr)ronly_differentiable op_inputss r _get_inputsrUs&ii))"-9525 r#c|xr| }t|s|g} tt|}t |t j rt|d}t|}n t|d}tjt|}tjt|}r'tdtj|z}fd} t|} t|} | rt} | D]} t| |D]D}||vr|j| vs| |js*| j!|jF|sZ| j"D]!}|| vs| |s| j!|#t%| | | } | r|s| D cgc] } | |vs|  } } | S#t$rgcYSwxYwcc} w)aDo a backward graph walk and return all the visited ops. Args: seed_ops: an iterable of operations from which the backward graph walk starts. If a list of tensors is given instead, the seed_ops are set to be the generators of those tensors. inclusive: if True the given seed_ops are also part of the resulting set. within_ops: an iterable of `tf.Operation` within which the search is restricted. If `within_ops` is `None`, the search is performed within the whole graph. within_ops_fn: if provided, a function on ops that should return True iff the op is within the graph traversal. This can be used along within_ops, in which case an op is within if it is also in within_ops. stop_at_ts: an iterable of tensors at which the graph walk stops. control_inputs: if True, control inputs will be used while moving backward. only_differentiable: if True, only traverse ops which are differentiable. This includes natively differentiable ops, or ops with custom gradients. Returns: A Python set of all the `tf.Operation` behind `seed_ops`. Raises: TypeError: if `seed_ops` or `within_ops` cannot be converted to a list of `tf.Operation`. FrHc8duxs|vxrduxs|SrQ)r within_ops within_ops_fns r is_withinz(get_backward_walk_ops..is_within!s1 $  2" "2 42r!24r#)rS)rnextr StopIterationrrrrFrIrOrObjectIdentitySetlistrrUrrcontrol_inputsr")seed_ops inclusiverYrZ stop_at_tsr`rS first_seed_opr,r[resultwavenew_wavernew_tnew_ops `` rget_backward_walk_opsrjs<"?+>'>. X zHh(M  z001 e 4B!"%HxU;H00 1KL*  . .x/H I( ?J 22:>J H4 >& X$ uH !r7JK!% J   886 !i&9 ,,uxx ! '' !F 6 !i&7 LL  ! !vx( D  ! 8RRx%7b 8F 8 -O  IL9sF1 G)G1 F?>F?ceZdZdZdZy)UnliftableErrorz3Raised if a Tensor cannot be lifted from the graph.TN)__name__ __module__ __qualname____doc__ag_pass_throughrXr#rrlrl:s ;/r#rlcRt|tjr |jS|SrQ)rrrr) op_or_tensors r _as_operationrtAs" j//0 ?? r#c|jDcgc]}|jc}t|jzScc}wrQ)rRrr_r`)rxs r graph_inputsrwGs. "1!$$ "T"*;*;%< << "s;c8t|tjr |j}t|ts|g}|Dcgc] }t |}}t d|D}t |}i}|r|j}||vr|j|||k(rE|} | g} | |vr|| } | j| | |vrdjdt| DSt|D]"} | |vs| |vs ||| <|j| $|rycc}w)a$Find one path from `from_op` to any of `tensors`, ignoring `sources`. Args: from_op: A `tf.Operation`. tensors: A `tf.Operation`, a `tf.Tensor`, or a list thereof. sources: A list of `tf.Tensor`. Returns: A python string containing the path, or "??" if none is found. c34K|]}|jywrQr .0rvs r zshow_path..^s*QADD*sz <- c3TK|] }|jd|jd"yw)z ()N)r r(rzs rr|zshow_path..lsN!affaff5Ns&(z??) rrrrr_rtrpoprrr0reversedrw) from_optensorssourcesr final_ops visited_ops ops_to_visitsome_op_outputrpath_oppathinps r show_pathrKs,**+jjG GT "iG3:;}V$;);*'**+i,.    B [OOB W}gYd 9 $ ) G 9 $[[Nx~N NNb!## k !c&8 ".    c "# " -rs*+<2 & $# L+"?< &@@%)%)(,%')..3 L^i ='X3r#