o eb@sdZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z e d gd Z dd d ZddZddZ  dddZe jdddZdS)z&Utilities for using generic resources.N)dtypes)ops) array_ops)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)rrZadd_to_collection GraphKeys RESOURCESLOCAL_RESOURCES)r Z create_opZis_initialized_opZ is_sharedresourcerOD:\Projects\ConvertPro\env\Lib\site-packages\tensorflow/python/ops/resources.pyregister_resource"s rcCttjjS)z6Returns resources visible to all tasks in the cluster.)rget_collectionr r rrrrshared_resources8rcCr)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)rsrrrrbs)rrrZ name_scopeosenvirongetZdevicerZconstantrstringrZ logical_notstackZ boolean_mask) resource_listrZ local_deviceZvariables_maskZvariable_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. cSrr)r rrrrrsrz(initialize_resources..r)r)rgroupZno_op)r&rrrrinitialize_resourcesgs  r))T)Nr)r')__doc__ collectionsr!Ztensorflow.python.frameworkrrZtensorflow.python.opsrrrZtensorflow.python.utilr namedtuplerrrrrZshould_use_resultr)rrrrs(        %