diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py index 36561a8c301..b1e0e162278 100644 --- a/tensorflow/python/framework/ops.py +++ b/tensorflow/python/framework/ops.py @@ -5875,7 +5875,7 @@ def dismantle_graph(graph): graph.__dict__ = {} -@tf_export("add_to_collection") +@tf_export(v1=["add_to_collection"]) def add_to_collection(name, value): """Wrapper for `Graph.add_to_collection()` using the default graph. @@ -5894,7 +5894,8 @@ def add_to_collection(name, value): """ get_default_graph().add_to_collection(name, value) -@tf_export("add_to_collections") + +@tf_export(v1=["add_to_collections"]) def add_to_collections(names, value): """Wrapper for `Graph.add_to_collections()` using the default graph. diff --git a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt index 17dc2003cef..a850f0415d4 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt @@ -508,14 +508,6 @@ tf_module { name: "add_n" argspec: "args=[\'inputs\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " } - member_method { - name: "add_to_collection" - argspec: "args=[\'name\', \'value\'], varargs=None, keywords=None, defaults=None" - } - member_method { - name: "add_to_collections" - argspec: "args=[\'names\', \'value\'], varargs=None, keywords=None, defaults=None" - } member_method { name: "arg_max" argspec: "args=[\'input\', \'dimension\', \'output_type\', \'name\'], varargs=None, keywords=None, defaults=[\"\", \'None\'], " diff --git a/tensorflow/tools/compatibility/renames_v2.py b/tensorflow/tools/compatibility/renames_v2.py index 49dd5e7ad49..9bb3276b76c 100644 --- a/tensorflow/tools/compatibility/renames_v2.py +++ b/tensorflow/tools/compatibility/renames_v2.py @@ -56,6 +56,8 @@ renames = { 'tf.VariableScope': 'tf.compat.v1.VariableScope', 'tf.WholeFileReader': 'tf.compat.v1.WholeFileReader', 'tf.accumulate_n': 'tf.math.accumulate_n', + 'tf.add_to_collection': 'tf.compat.v1.add_to_collection', + 'tf.add_to_collections': 'tf.compat.v1.add_to_collections', 'tf.all_variables': 'tf.compat.v1.all_variables', 'tf.angle': 'tf.math.angle', 'tf.assert_greater_equal': 'tf.debugging.assert_greater_equal',