aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/training/training.py
diff options
context:
space:
mode:
authorGravatar Asim Shankar <ashankar@google.com>2018-03-05 17:28:12 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-05 17:32:21 -0800
commitd576afdcd38dcfd9d0f6ce6d6cb262d22e2b11dd (patch)
treed61e36692a0491f428542bc0de297b6b0d950447 /tensorflow/python/training/training.py
parent665a4bf664546224c65eeb5a0a52d80e48e2f3e1 (diff)
gradients: Export tf.custom_gradients
(Moved from the tf.contrib.eager namespace) PiperOrigin-RevId: 187950503
Diffstat (limited to 'tensorflow/python/training/training.py')
-rw-r--r--tensorflow/python/training/training.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/training/training.py b/tensorflow/python/training/training.py
index 78c8ce9208..e623e27a21 100644
--- a/tensorflow/python/training/training.py
+++ b/tensorflow/python/training/training.py
@@ -28,6 +28,7 @@ See the @{$python/train} guide.
@@ProximalGradientDescentOptimizer
@@ProximalAdagradOptimizer
@@RMSPropOptimizer
+@@custom_gradient
@@gradients
@@AggregationMethod
@@stop_gradient