BVhU FdZddlZddlmZddlmZdZdZGdde Z y) zRegistry mechanism for "registering" classes/functions for general use. This is typically used with a decorator that calls Register for adding a class or function to a registry. N) tf_logging)compatlocationtypec2eZdZdZddgZdZd dZdZdZy) Registryz'Provides a registry for saving objects._name _registryc ||_i|_y)zCreates a new registry.N)r r selfnames T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/registry.py__init__zRegistry.__init__&sDJDNNc|s |j}||jvrQ|j|t}td|j||j |j |jfztjdd|||jtjd}tdt|dz }|dk\r||}nd}t|t|i|j|<y ) aERegisters a Python object "candidate" for the given "name". Args: candidate: The candidate object to add to the registry. name: An optional string specifying the registry key for the candidate. If None, candidate.__name__ will be used. Raises: KeyError: If same name is used twice. zJRegistering two %s with name '%s'! (Previous registration was in %s %s:%d)zRegistering %s (%s) in %s.)limitr)UNKNOWNrrrrN)__name__r _LOCATION_TAGKeyErrorr rfilenamelinenologgingvlog traceback extract_stackminlen _TYPE_TAG)r candidaterframestack stack_index location_tags rregisterzRegistry.register+s    d t~~nnT"=1e  4 ::tUZZ F G HH  LL0$ 4::N  # #! ,EaUa(Ka;'lLl%y-NDNN4rc6|jjS)zWLists registered items. Returns: A list of names of registered objects. )r keys)r s rlistz Registry.listIs >>    rctj|}||jvr|j|tSt |j d|)zLooks up "name". Args: name: a string specifying the registry key for the candidate. Returns: Registered object if found Raises: LookupError: if "name" has not been registered. z registry has no entry for: )ras_strr r# LookupErrorr r s rlookupzRegistry.lookupQsM == D t~~ ^^D !) ,, 04 D A CCr)N) r __module__ __qualname____doc__ __slots__rr)r,r0rrrr!s%/ $) O<!Crr) r3rtensorflow.python.platformrrtensorflow.python.utilrrr#objectrr5rrr9s/  <)  ?Cv?Cr