From 7cd00438e70519563c4ce5efb952d6a4b7b18d17 Mon Sep 17 00:00:00 2001
From: Karl Lattimer <karl@qdh.org.uk>
Date: Mon, 31 Oct 2016 17:06:39 +0000
Subject: [PATCH] Added list functionality to registry object (#4946)
* Added list functionality to registry object
* Fix indentation
* Punctuation fixes for comments.
---
tensorflow/python/framework/registry.py | 8 ++++++++
1 file changed, 8 insertions(+)
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".