Fixes reference to tf.contrib.defun in the init_scope documentation.
PiperOrigin-RevId: 277965575
Change-Id: I16a6f8173a2ca18fe0c64298e16cce28df8de261
diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py
index abb9620..9f0ddef 100644
--- a/tensorflow/python/framework/ops.py
+++ b/tensorflow/python/framework/ops.py
@@ -5456,15 +5456,14 @@
(3) The gradient tape is paused while the scope is active.
When eager execution is enabled, code inside an init_scope block runs with
- eager execution enabled even when defining graph functions via
- tf.contrib.eager.defun. For example:
+ eager execution enabled even when tracing a `tf.function`. For example:
```python
tf.compat.v1.enable_eager_execution()
- @tf.contrib.eager.defun
+ @tf.function
def func():
- # A defun-decorated function constructs TensorFlow graphs,
+ # A function constructs TensorFlow graphs,
# it does not execute eagerly.
assert not tf.executing_eagerly()
with tf.init_scope():
