o 2hT@sddZddlZddlZddlmZddlmZddlmZddlm Z e dgdGd d d e Z dS) z!Manages a Trackable object graph.N)base) converter)object_identity) tf_exportztrain.TrackableView)v1c@sHeZdZdZddZeejjfddZ e ddZ dd Z d d Z d S) TrackableViewaGathers and serializes a trackable view. Example usage: >>> class SimpleModule(tf.Module): ... def __init__(self, name=None): ... super().__init__(name=name) ... self.a_var = tf.Variable(5.0) ... self.b_var = tf.Variable(4.0) ... self.vars = [tf.Variable(1.0), tf.Variable(2.0)] >>> root = SimpleModule(name="root") >>> root.leaf = SimpleModule(name="leaf") >>> trackable_view = tf.train.TrackableView(root) Pass root to tf.train.TrackableView.children() to get the dictionary of all children directly linked to root by name. >>> trackable_view_children = trackable_view.children(root) >>> for item in trackable_view_children.items(): ... print(item) ('a_var', ) ('b_var', ) ('vars', ListWrapper([, ])) ('leaf', ...) cCs&t|tjr ||_dSt||_dS)zConfigure the trackable view. Args: root: A `Trackable` object whose variables (including the variables of dependencies, recursively) should be saved. May be a weak reference. N) isinstanceweakrefref _root_ref)selfrootrf/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/checkpoint/trackable_view.py__init__7s zTrackableView.__init__cKsF|i}|j|fi|D]\}}tj||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. )parent)_maybe_initialize_trackable_trackable_childrenitemsrconvert_to_trackable)clsobj save_typekwargschildrennamer rrrrEs   zTrackableView.childrencCs,t|jtjr|}|dusJ|S|jS)N)rr r r )r derefedrrrr Ys  zTrackableView.rootcCs |dS)zKReturns a list of all nodes from self.root using a breadth first traversal.r)_descendants_with_paths)r rrr descendantsbs zTrackableView.descendantscCsg}t|jg}t}d||j<|rA|}||||D]\}}||vr>||t ||f||<||q$|s||fS)zYReturns a list of all nodes and its paths from self.root using a breadth first traversal.r) collectionsdequer rObjectIdentityDictionarypopleftappendrrrTrackableReference)r bfs_sortedto_visit node_pathscurrent_trackabler dependencyrrrrfs"     z%TrackableView._descendants_with_pathsN)__name__ __module__ __qualname____doc__r classmethodrSaveType CHECKPOINTrpropertyr rrrrrrrs  r) r-rr tensorflow.python.trackablerrtensorflow.python.utilr tensorflow.python.util.tf_exportrobjectrrrrrs