diff --git a/tensorflow/python/framework/registry.py b/tensorflow/python/framework/registry.py index 16fb6eff26a..2e45acb4995 100644 --- a/tensorflow/python/framework/registry.py +++ b/tensorflow/python/framework/registry.py @@ -67,6 +67,14 @@ class Registry(object): stack = traceback.extract_stack() self._registry[name] = {_TYPE_TAG: candidate, _LOCATION_TAG: stack[2]} + def list(self): + """Lists registered items. + + Returns: + A list of names of registered objects. + """ + return self._registry.keys() + def lookup(self, name): """Looks up "name".