o ?e&@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 rZadd_to_collection GraphKeys RESOURCESLOCAL_RESOURCES)r Z create_opZis_initialized_opZ is_sharedresourcer`/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/resources.pyregister_resource#s rcCttjjS)z6Returns resources visible to all tasks in the cluster.)rget_collectionr rrrrrshared_resources9rcCr)z7Returns resources intended to be local to this session.)rrr rrrrrlocal_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. NZ.TF_DEVICE_FOR_UNINITIALIZED_VARIABLE_REPORTINGz/cpu:0)ZdtypecSg|]}|jqSr)r .0rrrr `z2report_uninitialized_resources..cSsg|]}|jjqSr)r name)rsrrrrcs)rrrZ name_scopeosenvirongetZdevicerZconstantrstringrZ logical_notrstackZ boolean_mask) resource_listr Z local_deviceZvariables_maskZvariable_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. cSrr)r rrrrrtrz(initialize_resources..r )r )rgroupZno_op)r'r rrrinitialize_resourceshs  r*)T)Nr)r()__doc__ collectionsr"Ztensorflow.python.frameworkrrZtensorflow.python.opsrrrrZtensorflow.python.utilr namedtupler rrrrZshould_use_resultr*rrrrs*         %