o 2hD*@sdZddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZgaGd d d Ze jd d ZddZGdddeZGddde jedZedGdddeZGdddeZdS)z7Definitions for resource-type trackable object classes.N)context) def_function)ops)tensor)base) tf_contextlib) tf_exportc@s2eZdZdZdgZddZeddZddZd S) ResourceTrackerz*An object that tracks a list of resources. _resourcescCs g|_dSNr selfr_/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/trackable/resource.py__init__&s zResourceTracker.__init__cC|jSr r r rrr resources)zResourceTracker.resourcescCs|j|dSr )r append)rresourcerrr add_resource-szResourceTracker.add_resourceN) __name__ __module__ __qualname____doc__ __slots__rpropertyrrrrrrr !s  r ccs,tt}t|zdVW|adS|aw)aA context to manage resource trackers. Use this in order to collect up all resources created within a block of code. Example usage: ```python resource_tracker = ResourceTracker() with resource_tracker_scope(resource_tracker): resource = TrackableResource() assert resource_tracker.resources == [resource] Args: resource_tracker: The passed in ResourceTracker object Yields: A scope in which the resource_tracker is active. N)list_RESOURCE_TRACKER_STACKr)resource_trackeroldrrrresource_tracker_scope1s  r"csfdd}|S)z"To avoid capturing loop variables.csg|Ri|Sr r)argskwargscaptured_gettercaptured_previousrrgetterQsz_make_getter..getterr)r&r'r(rr%r _make_getterNsr)c@seZdZdZddZdS)_ResourceMetaclassz!Metaclass for CapturableResource.csLfddfdd}tj}|D]}t||}q||i|S)Ncs8|dusJjg|Ri|}|j|i||Sr )__new__r) next_creatorakwobjclsrrdefault_resource_creator\s z=_ResourceMetaclass.__call__..default_resource_creatorcsdg|Ri|Sr r)r-r.)r2rrbsz-_ResourceMetaclass.__call__..)rget_default_graph_resource_creator_stack_resource_typer))r1r#r$previous_getterresource_creator_stackr(r)r1r2r__call__Zs    z_ResourceMetaclass.__call__N)rrrrr9rrrrr*Ws r*cseZdZdZdddZeddZeddZej d dZd d Z ed d Z e j dd Z ddZ ddZ eddZddZejjffdd ZddZZS)CapturableResourceazHolds a Tensor which a tf.function can capture. `CapturableResource`s are discovered by traversing the graph of object attributes, e.g. during `tf.saved_model.save`. They are excluded from the scope-based tracking of `TrackableResource`; generally things that require initialization should inherit from `TrackableResource` instead of `CapturableResource` directly. cCs0d|_||_trtj|_dStj|_dS)adInitialize the `CapturableResource`. Args: device: A string indicating a required placement for this resource, e.g. "CPU" if this resource must be created on a CPU device. A blank device allows the user to place resource creation, so generally this should be blank unless the resource only makes sense on one device. N) _resource_handle_value_resource_devicerexecuting_eagerly eager_moderr4 as_default_self_destruction_contextrdevicerrrrts   zCapturableResource.__init__cCrr )rr0rrrr6rz!CapturableResource._resource_typecCst|dtjS)NrA)getattr contextlibsuppressr rrr_destruction_contextsz'CapturableResource._destruction_contextcCs ||_dSr )rA)rdestruction_contextrrrrGs cCstd)z*A function that creates a resource handle.z3TrackableResource._create_resource not implemented.)NotImplementedErrorr rrr_create_resourcesz#CapturableResource._create_resourcecCrr )r<r rrr_resource_handlerz#CapturableResource._resource_handlecCs(t|tjtjfrt||_||_dSr ) isinstancerTensorr EagerTensorweakrefref_parent_trackabler<)rvaluerrrrKs  cCdS)z3A function that initializes the resource. Optional.Nrr rrr _initializezCapturableResource._initializecCrS)z0A function that destroys the resource. Optional.Nrr rrr_destroy_resourcerUz$CapturableResource._destroy_resourcecCsJ|jdur"t|j||_Wd|jS1swY|jS)z:Returns the resource handle associated with this Resource.N)rKrrCr=rJr rrrresource_handles   z"CapturableResource.resource_handlecKs^t|}t|j |}Wdn1swY||_|||<|||j<|jgS)zFor implementing `Trackable`.N)copyrrCr=rJrKrW)r object_map tensor_map unused_kwargsnew_obj new_resourcerrr_export_to_saved_model_graphs   z/CapturableResource._export_to_saved_model_graphc stj|fi|}|dkr>tjgddfdd}tjgddfdd}tjgddfdd }||||d |S) N savedmodelF)input_signature autographcs }|Sr )rJ)rr rr_creatorz8CapturableResource._trackable_children.._creatorc dSN)rTrr rr _initializerrcz._initializercrdre)rVrr rr _destroyerrcz:CapturableResource._trackable_children.._destroyer)rJrTrV)super_trackable_childrenrfunctionupdate)r save_typer$childrenrbrgrh __class__r rrjs   z&CapturableResource._trackable_childrencCsPz||WdWdS1swYWdSty'YdSwr )rGrV Exceptionr rrr__del__s  & zCapturableResource.__del__r;)rrrrr classmethodr6rrGsetterrJrKrTrVrWr^rSaveType CHECKPOINTrjrr __classcell__rrrorr:js*        r:) metaclassz*saved_model.experimental.TrackableResourcecs"eZdZdZdfdd ZZS)TrackableResourceaiHolds a Tensor which a tf.function can capture. A TrackableResource is most useful for stateful Tensors that require initialization, such as `tf.lookup.StaticHashTable`. `TrackableResource`s are discovered by traversing the graph of object attributes, e.g. during `tf.saved_model.save`. A TrackableResource has three methods to override: * `_create_resource` should create the resource tensor handle. * `_initialize` should initialize the resource held at `self.resource_handle`. * `_destroy_resource` is called upon a `TrackableResource`'s destruction and should decrement the resource's ref count. For most resources, this should be done with a call to `tf.raw_ops.DestroyResourceOp`. Example usage: >>> class DemoResource(tf.saved_model.experimental.TrackableResource): ... def __init__(self): ... super().__init__() ... self._initialize() ... def _create_resource(self): ... return tf.raw_ops.VarHandleOp(dtype=tf.float32, shape=[2]) ... def _initialize(self): ... tf.raw_ops.AssignVariableOp( ... resource=self.resource_handle, value=tf.ones([2])) ... def _destroy_resource(self): ... tf.raw_ops.DestroyResourceOp(resource=self.resource_handle) >>> class DemoModule(tf.Module): ... def __init__(self): ... self.resource = DemoResource() ... def increment(self, tensor): ... return tensor + tf.raw_ops.ReadVariableOp( ... resource=self.resource.resource_handle, dtype=tf.float32) >>> demo = DemoModule() >>> demo.increment([5, 1]) r;cs&tD]}||qtj|ddS)acInitialize the `TrackableResource`. Args: device: A string indicating a required placement for this resource, e.g. "CPU" if this resource must be created on a CPU device. A blank device allows the user to place resource creation, so generally this should be blank unless the resource only makes sense on one device. rCN)rrrir)rrCr rorrrs zTrackableResource.__init__rs)rrrrrrxrrrorrzs'rzcs6eZdZdZd fdd ZeddZddZZS) RestoredResourcezRestored SavedResource.r;cstj|ddS)Nr{)rirrBrorrr%szRestoredResource.__init__cKs*||jjd}|d}|dur||_|S)Nr{rJ)rrCgetrJ)r1 object_proto dependenciesr[r/resource_creatorrrr_deserialize_from_proto(s  z(RestoredResource._deserialize_from_protocCs<t|||t|tjrt|tjs|||dSdSdSr )setattrrLr TrackablerFunction_track_trackable)rnamerRrrr_add_trackable_child0s   z%RestoredResource._add_trackable_childrs) rrrrrrtrrrxrrrorr|"s  r|)rrErXrOtensorflow.python.eagerrrtensorflow.python.frameworkrrtensorflow.python.trackablertensorflow.python.utilr tensorflow.python.util.tf_exportrrr contextmanagerr"r)typer*rr:rzr|rrrrs*         8