a /Sic@sdZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z e d gd Z dd d ZddZddZdddZe jdddZdS)z&Utilities for using generic resources.N)dtypes)ops) array_ops)control_flow_ops)math_ops) tf_should_use _Resource)handlecreateis_initializedTcCs6t|||}|r"ttjj|nttjj|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)rradd_to_collection GraphKeys RESOURCESLOCAL_RESOURCES)r create_opis_initialized_op is_sharedresourcer[/var/www/html/django/DPS/env/lib/python3.9/site-packages/tensorflow/python/ops/resources.pyregister_resource"s rcCsttjjS)z6Returns resources visible to all tasks in the cluster.)rget_collectionr rrrrrshared_resources8srcCsttjjS)z7Returns resources intended to be local to this session.)rrr rrrrrlocal_resources=srreport_uninitialized_resourcesc Cs|durtt}t|tjdd}t||sjtj gt j dWdWdSt tdd|D}t dd|D}t||WdWdS1s0YWdn1s0YdS)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)dtypecSsg|] }|jqSr)r .0rrrr _z2report_uninitialized_resources..cSsg|] }|jjqSr)r name)rsrrrr br!)rrr name_scopeosenvirongetdevicerconstantrstringr logical_notstack boolean_mask) resource_listr" local_devicevariables_maskvariable_names_tensorrrrrBs    , 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. cSsg|] }|jqSr)r rrrrr sr!z(initialize_resources..r")r")rgroupno_op)r.r"rrrinitialize_resourcesgs r5)T)Nr)r2)__doc__ collectionsr%tensorflow.python.frameworkrrtensorflow.python.opsrrrtensorflow.python.utilr namedtuplerrrrrshould_use_resultr5rrrrs&        %