Internally rewrite @recompute_grad to use @custom_gradient
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 27 Apr 2018 18:41:21 +0000 (11:41 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 27 Apr 2018 18:44:18 +0000 (11:44 -0700)
commitd1e0a73577b226d2a865a96f1b4ea9f463f3f4ed
tree923460dd919c2ec0161fa06f25f7785e494d655f
parentf7f02482f486a7c430fe030d62f756685cd8d9d0
Internally rewrite @recompute_grad to use @custom_gradient

PiperOrigin-RevId: 194571125
tensorflow/contrib/layers/python/layers/rev_block_lib.py
tensorflow/contrib/layers/python/layers/rev_block_lib_test.py
tensorflow/python/ops/custom_gradient.py
tensorflow/python/ops/gradients_test.py