tf.recompute_grad

View source on GitHub

An eager-compatible version of recompute_grad.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.recompute_grad

tf.recompute_grad(
    f
)

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.