tf.recompute_grad

View source on GitHub

An eager-compatible version of recompute_grad.

For f(*args, **kwargs), this supports gradients with respect to args or kwargs, but kwargs are currently only supported in eager-mode. Note that for keras layer and model objects, this is handled automatically.

Args
f function f(*x) that returns a Tensor or sequence of Tensor outputs.
Returns
A function g that wraps f, but which recomputes f on the backwards pass of a gradient call.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/recompute_grad