Add migration blocks to tf.compat.v1.get_default_graph and reset_default_graph docstrings
PiperOrigin-RevId: 387186958
Change-Id: I3e0a160bec4a1916f273ab16d2b84e272b6d22f5
diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py
index 1086b52..eed2702 100644
--- a/tensorflow/python/framework/ops.py
+++ b/tensorflow/python/framework/ops.py
@@ -6126,6 +6126,16 @@
result in undefined
behavior. Using any previously created `tf.Operation` or `tf.Tensor` objects
after calling this function will result in undefined behavior.
+
+ @compatibility(TF2)
+ `reset_default_graph` does not work with either eager execution or
+ `tf.function`, and you should not invoke it directly. To migrate code that
+ uses Graph-related functions to TF2, rewrite the code without them. See the
+ [migration guide](https://www.tensorflow.org/guide/migrate) for more
+ description about the behavior and semantic changes between Tensorflow 1 and
+ Tensorflow 2.
+ @end_compatibility
+
Raises:
AssertionError: If this function is called within a nested graph.
"""
@@ -6149,6 +6159,15 @@
thread, you must explicitly add a `with g.as_default():` in that
thread's function.
+ @compatibility(TF2)
+ `get_default_graph` does not work with either eager execution or
+ `tf.function`, and you should not invoke it directly. To migrate code that
+ uses Graph-related functions to TF2, rewrite the code without them. See the
+ [migration guide](https://www.tensorflow.org/guide/migrate) for more
+ description about the behavior and semantic changes between Tensorflow 1 and
+ Tensorflow 2.
+ @end_compatibility
+
Returns:
The default `Graph` being used in the current thread.
"""
