BVhandZddlmZddlmZddlmZddlmZddlm Z dZ dZ d Z d Z dd Zdd Zy )z6Utilities for extracting and writing checkpoint info`.)trackable_object_graph_pb2)resource_variable_ops) variables)trackable_utils)object_identityct|}tj}|D] }dt|vs|j }|D]}t |D]\}} |j | |} | | j| jr td| |vrtd| dtj|| |||} | || <t|} | || <|j| t j"j$j'||| } |j)|gj| #|S#ttf$rd} YwxYw)zGather and name slot variables.get_slot_namesNzCurrently only variables with no dependencies can be saved as slot variables. File a feature request if this limitation bothers you.zCA slot variable was re-used as a dependency of a Trackable object: zW. This is not currently allowed. File a feature request if this limitation bothers you.) variable_pathoptimizer_path slot_name)r original_variable_node_idslot_variable_node_id)listrObjectIdentityDictionarydirr enumerateget_slotAttributeErrorKeyError_maybe_initialize_trackable_trackable_childrenNotImplementedErrorrslot_variable_keylenappendrTrackableObjectGraphTrackableObjectSlotVariableReference setdefault)trackable_objectsnode_ids object_namesnon_slot_objectsslot_variables trackable slot_namesr r original_variable slot_variablecheckpoint_namerslot_variable_protos Q/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/checkpoint/util.pyserialize_slot_variablesr,s+,";;=.#)Oi3y>)++-j!#O)>)Y //ct|tjstj|syt |dd|j jS|jS)zGGets the full name of variable for name-based checkpoint compatibility._save_slice_infoN) isinstancerVariableris_resource_variablegetattrr4 full_name _shared_name)vars r+ get_full_namer<QsS c9-- . 5 5c :  S$d+7    ) ))   r1ci}tj}t}t|jD]\}}|j s|j s|jdr|j||jD]3}|j}||vr t||<||j|5t}|j||rW|j}||vr|j|} |j| | D]} | |vs|j| |rWt|jD]!\}}t||v|j_#y)zDetermines which objects have checkpoint values and save this to the proto. Args: object_graph_proto: A `TrackableObjectGraph` proto. registered_saverN)rObjectIdentitySetsetrnodes attributesr$HasFieldaddchildrennode_idupdatepopboolhas_checkpoint_valuesvalue) object_graph_protoparentscheckpointed_trackablesrF object_proto child_protochildto_visitr%current_parentsparents r+add_checkpoint_values_checkrUasf '+==? E();)A)AB"g|<#>#>01!!'*#,," !!e g  en! " "U( //)* Ikk),O""?3! 7  V  ));)A)AB,g|/3**0,L&&,,r1c`|j\}}tj}|jD]\}}t j |||<tj}t |D] \}} ||| < |rtj} nt|||} |||| |fS)a]Traverse the object graph and list all accessible objects. Looks for `Trackable` objects which are dependencies of `root_trackable`. Includes slot variables only if the variable they are slotting for and the optimizer are dependencies of `root_trackable` (i.e. if they would be saved with a checkpoint). Args: graph_view: A GraphView object. skip_slot_variables: If True does not return trackables for slot variable. Default False. Returns: A tuple of (trackable objects, paths from root for each object, object -> node id, slot variables, object_names) )r r!r")breadth_first_traversalrritemsrobject_path_to_stringrr,) graph_viewskip_slot_variablesr node_pathsr"objpathr!rFnoder$s r+(objects_ids_and_slot_variables_and_pathsr`s$#-"D"D"FZ 99;,##%Dic4'==dCLD  5 5 7( !23mgtHTN$==?N-+!N Z>< PPr1c$t||d}|S)z:Traverse the object graph and list all accessible objects.r)r`)rZr[r s r+ list_objectsrbs#>% r1N)F)__doc__tensorflow.core.protobufrtensorflow.python.opsrrtensorflow.python.trackablertensorflow.python.utilrr,r0r<rUr`rbr1r+risB=?7+72.b0  *,\BG!QHr1