AVh&dZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z ejd gd Zdd Zd ZdZ ddZe j&ddZy)z&Utilities for using generic resources.N)dtypes)ops) array_ops)array_ops_stack)control_flow_ops)math_ops) tf_should_use _Resource)handlecreateis_initializedct|||}|r/tjtjj|ytjtjj |y)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_sharedresources O/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/resources.pyregister_resourcer#sJvy*; <(#--118<#--77Bc\tjtjjS)z6Returns resources visible to all tasks in the cluster.)rget_collectionrrrrshared_resourcesr9s  CMM33 44rc\tjtjjS)z7Returns resources intended to be local to this session.)rrrrrrrlocal_resourcesr>s  CMM99 ::rc |ttz}tj|5tj j dd}tj|5|s7tjgtjcdddcdddStjtj|Dcgc]}|j c}}tj|Dcgc]}|j"j$c}}tj&||cdddcdddScc}wcc}w#1swYnxYw dddy#1swYyxYw)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)dtype)rrr name_scopeosenvirongetdevicerconstantrstringr logical_notrstackr r name boolean_mask) resource_listr, local_devicervariables_masksvariable_names_tensors rreport_uninitialized_resourcesr4Cs> $&)::M ~~dK::>>8(DL L ! K !!"FMM: K K KK ++   = Ia!1!1 I JLn(00"/ 0Q188== 02 # #$9> J K K KK!J 1 K K K KKKsM6E#'E E"E?D<E0E E) E< EE EE%c|r-tj|Dcgc]}|jc}d|iStj|Scc}w)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. r,)r,)rgroupr no_op)r.r,r0s rinitialize_resourcesr8hsC  ! !m#DAHH#D P4 PP   T **$EsA)T)Nr4)init)__doc__ collectionsr$tensorflow.python.frameworkrrtensorflow.python.opsrrrrtensorflow.python.utilr namedtupler rrrr4should_use_resultr8rrrrAsx"- .++12*0 #K " ";#I K C,5 ; 26(H"KJ   +! +r