BVhTxdZddlZddlZddlmZddlmZddlmZddlm Z e dgGd d e Z y) z!Manages a Trackable object graph.N)base) converter)object_identity) tf_exportztrain.TrackableView)v1cneZdZdZdZeejjfdZ e dZ dZ dZ y) 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', ...) c|t|tjr||_ytj||_y)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)selfroots [/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/checkpoint/trackable_view.py__init__zTrackableView.__init__7s+)w{{;dDN";;t, Nc |ji}|j|fi|jD]!\}}tj||}|||<#|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 s rrzTrackableView.childrenEse##%H,S,,YA&AGGI c  * *3s ;chtn Orct|jtjr|j}|J|S|jS)N)r rr r )rderefeds rrzTrackableView.rootYs;$..'++. g   n ^^rc(|jdS)zKReturns a list of all nodes from self.root using a breadth first traversal.r)_descendants_with_paths)rs r descendantszTrackableView.descendantsbs  ' ' )! ,,rcg}tj|jg}tj}d||j<|r|j }|j ||j|jD];\}}||vs ||tj||fz||<|j |=|r||fS)zYReturns a list of all nodes and its paths from self.root using a breadth first traversal.) collectionsdequerrObjectIdentityDictionarypopleftappendrrrTrackableReference)r bfs_sortedto_visit node_pathscurrent_trackabler dependencys rr#z%TrackableView._descendants_with_pathsfsJ  $))-H 99;JJtyy "**,)*"mm,=>DDF& $ Z '*+&&tZ8:; Z  //* % &  z !!rN)__name__ __module__ __qualname____doc__r classmethodrSaveType CHECKPOINTrpropertyrr$r#r&rrr r sK8 .#'==#;#;&  -"rr ) r5r'r tensorflow.python.trackablerrtensorflow.python.utilr tensorflow.python.util.tf_exportrobjectr r&rrr>sB',126 R(["F[")["r