o 2h&@sdZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z e d gd Zdd dZddZddZ  dddZe jdddZdS)z&Utilities for using generic resources.N)dtypes)ops) array_ops)array_ops_stack)control_flow_ops)math_ops) tf_should_use _Resource)handlecreateis_initializedTcCs8t|||}|rttjj|dSttjj|dS)aRegisters a resource into the appropriate collections. This makes the resource findable in either the shared or local resources collection. Args: handle: op which returns a handle for the resource. create_op: op which initializes the resource. is_initialized_op: op which returns a scalar boolean tensor of whether the resource has been initialized. is_shared: if True, the resource gets added to the shared resource collection; otherwise it gets added to the local resource collection. N)r radd_to_collection GraphKeys RESOURCESLOCAL_RESOURCES)r create_opis_initialized_op is_sharedresourcerZ/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/ops/resources.pyregister_resource#s rcCttjjS)z6Returns resources visible to all tasks in the cluster.)rget_collectionrrrrrrshared_resources9rcCr)z7Returns resources intended to be local to this session.)rrrrrrrrlocal_resources>rrreport_uninitialized_resourcesc Cs|dur tt}t|atjdd}t|D|s5tj gt j dWdWdSt tdd|D}t dd|D}t||WdWdS1scwYWddS1sswYdS)aReturns the names of all uninitialized resources in resource_list. If the returned tensor is empty then all resources have been initialized. Args: resource_list: resources to check. If None, will use shared_resources() + local_resources(). name: name for the resource-checking op. Returns: Tensor containing names of the handles of all resources which have not yet been initialized. N.TF_DEVICE_FOR_UNINITIALIZED_VARIABLE_REPORTINGz/cpu:0)dtypecSg|]}|jqSr)r .0rrrr `z2report_uninitialized_resources..cSsg|]}|jjqSr)r name)r"srrrr$cs)rrr name_scopeosenvirongetdevicerconstantrstringr logical_notrstack boolean_mask) resource_listr& local_devicevariables_maskvariable_names_tensorrrrrCs,      "initcCs*|rtjdd|Dd|iStj|dS)zInitializes the resources in the given list. Args: resource_list: list of resources to initialize. name: name of the initialization op. Returns: op responsible for initializing all resources. cSr r)r r!rrrr$tr%z(initialize_resources..r&)r&)rgroupno_op)r2r&rrrinitialize_resourceshs  r9)T)Nr)r6)__doc__ collectionsr)tensorflow.python.frameworkrrtensorflow.python.opsrrrrtensorflow.python.utilr namedtupler rrrrshould_use_resultr9rrrrs*         %