diff options
| author |  Asim Shankar <ashankar@google.com> | 2018-03-05 17:28:12 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-05 17:32:21 -0800 |
| commit | d576afdcd38dcfd9d0f6ce6d6cb262d22e2b11dd (patch) | |
| tree | d61e36692a0491f428542bc0de297b6b0d950447 /tensorflow/python/eager/backprop_test.py | |
| parent | 665a4bf664546224c65eeb5a0a52d80e48e2f3e1 (diff) | |
gradients: Export tf.custom_gradients
(Moved from the tf.contrib.eager namespace)
PiperOrigin-RevId: 187950503
Diffstat (limited to 'tensorflow/python/eager/backprop_test.py')
| -rw-r--r-- | tensorflow/python/eager/backprop_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/eager/backprop_test.py b/tensorflow/python/eager/backprop_test.py index 48fd170764..07a2155d24 100644 --- a/tensorflow/python/eager/backprop_test.py +++ b/tensorflow/python/eager/backprop_test.py @@ -23,7 +23,6 @@ import numpy as np from tensorflow.python import pywrap_tensorflow from tensorflow.python.eager import backprop from tensorflow.python.eager import context -from tensorflow.python.eager import custom_gradient from tensorflow.python.eager import tape from tensorflow.python.eager import test from tensorflow.python.framework import constant_op @@ -32,6 +31,7 @@ from tensorflow.python.framework import ops from tensorflow.python.framework import tensor_shape from tensorflow.python.framework import test_util from tensorflow.python.ops import array_ops +from tensorflow.python.ops import custom_gradient from tensorflow.python.ops import embedding_ops from tensorflow.python.ops import gradients from tensorflow.python.ops import math_ops |