Add support for is_recompute optional kwarg to functions decorated with recompute_grad
authorA. Unique TensorFlower <gardener@tensorflow.org>
Thu, 24 May 2018 03:39:31 +0000 (20:39 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 24 May 2018 03:42:12 +0000 (20:42 -0700)
commit8f863f3d71542c47390f2d40348b72296ed5c4be
tree58adc3f4ea26c56c9416045981490b256da1c895
parent42e50daa384183d2f64e0ab5ae3f9bed07128e07
Add support for is_recompute optional kwarg to functions decorated with recompute_grad

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