From a2643a983694a91ef0027650bc0ce28f2f760067 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 12 Mar 2018 17:48:57 -0700 Subject: [PATCH] Make tf.add_to_collections visible. PiperOrigin-RevId: 188807786 --- tensorflow/python/framework/framework_lib.py | 1 + tensorflow/python/framework/ops.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/framework/framework_lib.py b/tensorflow/python/framework/framework_lib.py index 3172f3c..4bb030c 100644 --- a/tensorflow/python/framework/framework_lib.py +++ b/tensorflow/python/framework/framework_lib.py @@ -48,6 +48,7 @@ ## Graph collections @@add_to_collection +@@add_to_collections @@get_collection @@get_collection_ref @@GraphKeys diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py index f5dde3a..6174d32 100644 --- a/tensorflow/python/framework/ops.py +++ b/tensorflow/python/framework/ops.py @@ -5602,7 +5602,7 @@ def add_to_collection(name, value): """ get_default_graph().add_to_collection(name, value) - +@tf_export("add_to_collections") def add_to_collections(names, value): """Wrapper for `Graph.add_to_collections()` using the default graph. -- 2.7.4