BVhdZddlZddlZddlmZddlmZddlmZddlm Z e dgGd d ejZ y) z%Manages a graph of Trackable objects.N) save_util_v1)trackable_view)base) tf_exportz%__internal__.tracking.ObjectGraphView)v1ceZdZdZd dZfdZejjffd Z ejjfdZ e dZ e dZ dZfd Zd d Z d d ZxZS)ObjectGraphViewz'Gathers and serializes an object graph.ctjj||t|tj r|nt j ||_||_y)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_dependenciess W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/checkpoint/graph_view.pyr zObjectGraphView.__init__sG  ))$5 )w{{;d";;t, N"7Dc ~|j}|Atj||}tj||t |j<t |t|}||t |<t|jD]&\}}t||tj||(|SN) rcopydeepcopyrridsuper__new__typevarsitemssetattr)rmemo strong_root strong_copycopiedkeyvalue __class__s r __deepcopy__zObjectGraphView.__deepcopy__0s.."KMM+t4k!([!9d2dnn  W_T$Z (FDDN4j&&(7 U fc4==567 Mrc g}tt| ||fi|jD]*\}}|j t j ||,||jur'|jr|j|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. ) rr childrenr appendrTrackableReferencerrextend)robj save_typekwargsr+namerr(s r list_childrenzObjectGraphView.list_children?sH?**-yDrXsK+ 57,6 2r:MBn22MB;MBr