From 83739474b52a6691ed19132f646fc84f6198e51f Mon Sep 17 00:00:00 2001 From: Vishnuvardhan Janapati <46058173+jvishnuvardhan@users.noreply.github.com> Date: Mon, 10 May 2021 08:48:30 -0700 Subject: [PATCH] Update custom_gradient.py --- tensorflow/python/ops/custom_gradient.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/ops/custom_gradient.py b/tensorflow/python/ops/custom_gradient.py index 6c2a234db7a..2ab7608830e 100644 --- a/tensorflow/python/ops/custom_gradient.py +++ b/tensorflow/python/ops/custom_gradient.py @@ -203,8 +203,8 @@ def custom_gradient(f=None): gradient (as calculated by `tf.gradients`) is determined by `f(x)[1]`. Raises: - RuntimeError: tf.gradients is not supported when eager execution is - enabled. Use tf.GradientTape instead. + RuntimeError: `tf.gradients` is not supported when eager execution is + enabled. Use `tf.GradientTape` instead. """ if f is None: