Internally rewrite RevBlock to use @custom_gradient
authorA. Unique TensorFlower <gardener@tensorflow.org>
Sun, 29 Apr 2018 02:47:42 +0000 (19:47 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sun, 29 Apr 2018 02:50:17 +0000 (19:50 -0700)
commit3a9c513c3f4303e5194474d804367c1f4831e3ee
treefcbec0e200d3936fa015a06256b3bfcf7c0ecd82
parentd07a8d4071b20d10226ea81758c9306ffce21317
Internally rewrite RevBlock to use @custom_gradient

PiperOrigin-RevId: 194679657
tensorflow/contrib/layers/python/layers/rev_block_lib.py
tensorflow/contrib/layers/python/layers/rev_block_lib_test.py