o ?e@sfdZddlZddlZddlmZddlmZddlmZddlm Z e dgdGd d d ej Z dS) z%Manages a graph of Trackable objects.N) save_util_v1)trackable_view)base) tf_exportz%__internal__.tracking.ObjectGraphView)v1cseZdZdZdddZfddZejjffdd Z ejjfd d Z e d d Z e d dZ ddZfddZdddZ   dddZZS)ObjectGraphViewz'Gathers and serializes an object graph.NcCs4tj||t|tjr|nt||_||_dS)aConfigure the graph view. Args: root: A `Trackable` object whose variables (including the variables of dependencies, recursively) should be saved. May be a weak reference. attached_dependencies: List of dependencies to attach to the root object. Used when saving a Checkpoint with a defined root object. To avoid reference cycles, this should use the WeakTrackableReference class. N)r TrackableView__init__ isinstanceweakrefref _root_ref_attached_dependencies)selfrootattached_dependenciesrh/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/checkpoint/graph_view.pyr s   zObjectGraphView.__init__csz|}|durt||}t||t|j<tt|}||t|<t | D]\}}t ||t||q,|SN) r copydeepcopyr r idsuper__new__typevarsitemssetattr)rmemoZ strong_rootZ strong_copyZcopiedkeyvalue __class__rr __deepcopy__0s  zObjectGraphView.__deepcopy__c s^g}tt|j||fi|D] \}}|t||q||jur-|jr-| |j|S)a,Returns list of all child trackables attached to obj. Args: obj: A `Trackable` object. save_type: A string, can be 'savedmodel' or 'checkpoint'. **kwargs: kwargs to use when retrieving the object's children. Returns: List of all children attached to the object. ) rrchildrenrappendrZTrackableReferencerrextendrobj save_typekwargsr$namer r!rr list_children?s   zObjectGraphView.list_childrencKs,i}|j|fi|D]\}}|||<q |S)aAReturns all child trackables attached to obj. Args: obj: A `Trackable` object. save_type: A string, can be 'savedmodel' or 'checkpoint'. **kwargs: kwargs to use when retrieving the object's children. Returns: Dictionary of all children attached to the object with name to trackable. )r,r'rrrr$Us  zObjectGraphView.childrencCs|jS)a"Returns list of dependencies that should be saved in the checkpoint. These dependencies are not tracked by root, but are in the checkpoint. This is defined when the user creates a Checkpoint with both root and kwargs set. Returns: A list of TrackableReferences. )rrrrrres z%ObjectGraphView.attached_dependenciescCs,t|jtjr|}|dusJ|S|jSr)r r r r )rZderefedrrrrrs  zObjectGraphView.rootcCs|Sr)_breadth_first_traversalr-rrrbreadth_first_traversal{sz'ObjectGraphView.breadth_first_traversalcstt|S)z5Find shortest paths to all dependencies of self.root.)rrZ_descendants_with_pathsr-r!rrr.~sz(ObjectGraphView._breadth_first_traversalcCst||\}}}}|||fS)agDetermine checkpoint keys for variables and build a serialized graph. Non-slot variables are keyed based on a shortest path from the root saveable to the object which owns the variable (i.e. the one which called `Trackable._add_variable` to create it). Slot variables are keyed based on a shortest path to the variable being slotted for, a shortest path to their optimizer, and the slot name. Args: saveables_cache: An optional cache storing previously created SaveableObjects created for each Trackable. Maps Trackables to a dictionary of attribute names to Trackable. Returns: A tuple of (named_variables, object_graph_proto, feed_additions): named_variables: A dictionary mapping names to variable objects. object_graph_proto: A TrackableObjectGraph protocol buffer containing the serialized object graph and variable references. feed_additions: A dictionary mapping from Tensors to values which should be fed when saving. Raises: ValueError: If there are invalid characters in an optimizer's slot names. )rZ-serialize_object_graph_with_registered_savers)rZsaveables_cacheZnamed_saveable_objectsZobject_graph_protoZfeed_additions_rrrserialize_object_graphs  z&ObjectGraphView.serialize_object_graphcCst||||dS)z=Creates SaveableObjects with the current object graph frozen.r)rZfrozen_saveables_and_savers)rZ object_mapZto_graphZcall_with_mapped_capturesrrrfrozen_saveable_objectss z'ObjectGraphView.frozen_saveable_objectsr)NNN)__name__ __module__ __qualname____doc__r r#rZSaveType CHECKPOINTr,r$propertyrrr/r.r1r2 __classcell__rrr!rrs"       r) r6rr Ztensorflow.python.checkpointrrZtensorflow.python.trackablerZ tensorflow.python.util.tf_exportrrrrrrrs