BVh8.dZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd l m Z dd lmZe j Zd Zd Zej&dgdZej&dddgZdZdZedg ddZy)zUtility to lift subgraphs.N) func_graph)ops)tensor) array_ops) op_selector)resource_variable_ops)compat)object_identity) tf_exportcRt|tjr |jS|SN) isinstance tensor_libTensorop op_or_tensors U/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/lift_to_graph.py _as_operationr"s" j//0 ?? c^tdtjt|DS)Nc3|K|]4}t|jdk(xrt|j 6yw)ConstN)rtypecontrol_inputs).0is r z#_constant_inputs..)sE M1  " "h .5"1%4445 Ms:<)allr graph_inputsrrs r_constant_inputsr!(s0  M!..}\/JK M MMr_InputMutation copied_op input_indexold_graph_tensor_ControlMutationr$ old_graph_opcg}g}g}t|jD]s\}}|j|d} | Itjd|j |j } |jtd|||j| ug} |jD]D} |j| d} | |jtd| 4| j| Ftj| 5tj|j5|jj|jd} | ::nd3L***$$$$&l 4%0*8 :; &'" " 1 19!::& @A""#789  56 2998M!!"''40A}qvv.e6F6FFnnU!#,A,)+)9)9)?)?)A %3>>(+/0 J WW I$&* #%da!!!$F1I%+ ,   y  1 ,- .   y  1 . //- , , .sO< KB J4&J)9)J4"/J.J4%K-K  K) J44J= 9KK c|r0||vr,|j|||jj}nU|jjdk(rt |r|jj d}t |j|||\}} |s| rtdj|tj|jj5tj|||j|jj}dddnrtj|jj5tj|j|j|jj}dddt!j"|} | j$rt!j&| d ||<|j||j<y#1swYaxYw#1swYmxYw) aCreate a source in a graph based on a Tensor from a different graph. This function creates a placeholder analog of `s` in a graph with the following behavior: 1) If s is a captured Tensor or Variable and handle_captures is set to True, simply capture it in the new graph as well. 2) If s is a PlaceholderWithDefault whose default is a constant, preserve said default in the new graph. 3) When applicable, copy resource variable metadata from `s` to the newly created placeholder. Args: s: The source of interest. graph: The destination graph. op_map: A dict mapping ops and tensors in the old graph to the new one. handle_captures: A boolean indicating whether to re-capture s in the new graph or simply create a vanilla placeholder. inverse_captures: A dict mapping s back to the Tensor or Variable that it captures. base_graph: The graph being copied from. )r+PlaceholderWithDefaultrrrDrErFz4Could not copy source node {} because it has inputs.)inputr,r+N)r-r,r+T) graph_mode)capturerr+rr!r1rUAssertionErrorformatrr9rplaceholder_with_defaultr,r6r-rget_resource_handle_datashape_and_type_set_handle_shapes_and_types) srDrEhandle_capturesinverse_capturesrFcopied_placeholder default_valueunavailable_inputsunavailable_control_inputs base_handles r _copy_sourcerjs4..'7':Kttyy,,1A!1DDDKKNM5E   56227  @ 6-  "" ADDKK F$==}%QWW14499FFF ADDKK 8$00qwwQTTYY88&>>qA+66 !&)$&&&,%FF88s:G ?AG, G),G5z__internal__.lift_to_graph)v1c !g}g} |D]?} t| tjr|j| /| j| A|xs| dj}|xst j }t j|xsg}td|D} tjt} | D],} |jtj| ||| | |.g}tg!| Dcgc]}| t|s t|}}t|}|r|r|j}|!vr!j!||j|tj"|D]Q}|j$dk(r|j!|t'!fd| |Ds<||vsA|j|S|r|j)!|r#|jt+t-||r|j/dg}t j }g}t|t0j2rAt|t0j2r'|j4}|D] \}}|||< |j6}|j95|D]}|||< t}|D]D}||vs|j;||j!|j<t?||||||F|D].}|j!|j<t?||||||0g}g}tA|D]?}||vs||vr tC|||| \}}|jE||jE|A|jG5|D]5} | jHjK| jL|| jN7|D]D} | jPj$dk(r| jHjS|| jPF d d d |cd d d Scc}w#1swYxYw#1swYy xYw) aCopies the tensor and all its inputs recursively to the outer graph. Args: tensors: The Tensors to lift. graph: The graph to lift to. sources: Optional sequence of nodes to start from. If omitted the whole subgraph which feeds into `init_tensor` is lifted. disallowed_placeholders: An optional set of ops which may not appear in the lifted graph. Defaults to all placeholders. add_sources: A boolean indicating whether placeholders which are not in sources should be allowed. handle_captures: A boolean indicating whether to re-capture s in the new graph or simply create a vanilla placeholder. base_graph: The graph from which to lift ops. This will be inferred if not specified. op_map: A map contains all the existing nodes that have been lifted to the destination graph, so they won't be lifted and copied again. Returns: A mapping from ops in the current default graph to ops in `graph`. Raises: UnliftableError: If a placeholder blocks lifting. rc34K|]}|jywr r)rrPs rrz lift_to_graph..s*QADD*s) init_tensorsourcesdisallowed_placeholders visited_ops op_outputs add_sourcesTPUReplicateMetadatac3&K|]}|v ywr )rrP marked_opss rrz lift_to_graph..s9AZ9scDttj|dk(S)Nr)lenrr rns rzlift_to_graph..(s3{'?'?'C#D#Ir)rQ)rbrDrErcrdrFrXN)*rrResourceVariabler7rDr ObjectIdentityDictionaryObjectIdentitySetset collections defaultdictupdater map_subgraphrpopaddr rrdifference_updatenextitersortr FuncGraphcapturesinternal_captures as_defaultremoverrjreversedrUextend_mutation_lockr$ _update_inputr%r&r(_add_control_input)"tensorsrDrprqrtrcrFrEvariable_init_tensors init_tensorsrrrrsro ops_to_copyt ops_to_visit unvisited_opsrinprrdrexternal_captureinternal_capturer source_opsrbrGrHnew_input_mutationsnew_control_mutationsrTrxs" @r lift_to_graphrsB,"f&/@@A""6*&! " 2\!_22*  ?_==?&  - -gm <'*'**+&&s+*""k NN;++ 7 !""+2w*,87q#M!$45 "7,7l#-     b z nnR))"-## 88- - # 9C9 9 w    c "# ##J/$tM234+ 8IK($==?Z112 ,,-""H.6<**+;'(<"44  4 "fQiJ  # gqqtt+-!  # #!nnQTT )+ !!O{#6 z R6\3C uV 4D000145 6     M%E(((  &)B)B"C EE(M(  % %)? ? --fX5J5J.KL M M i44a7p M MQ44s2; P* P;?CP;BP/ P;/P8 4P;;Q)NNFFNN)__doc__rtensorflow.python.frameworkrrrrtensorflow.python.opsrrrtensorflow.python.utilr r tensorflow.python.util.tf_exportr UnliftableErrorrr! namedtupler"r'rUrjrrwrrrs !2+<+-7)26-- M(''46*;)).!# P/f9'x 'B/*.#"'!a0ar