tf.recompute

TensorFlow 1 version

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.

TensorFlow

tf

tf.audio

tf.autograph

tf.bitwise

tf.compat

tf.config

tf.data

tf.debugging

tf.distribute

tf.dtypes

tf.errors

tf.estimator

tf.experimental

tf.feature_column

tf.graph_util

tf.image

tf.initializers

tf.io

tf.keras

tf.linalg

tf.lite

tf.lookup

tf.losses

tf.math

tf.metrics

tf.nest

tf.nn

tf.optimizers

tf.quantization

tf.queue

tf.ragged

tf.random

tf.raw_ops

tf.saved_model

tf.sets

tf.signal

tf.sparse

tf.strings

tf.summary

tf.sysconfig

tf.test

tf.tpu

tf.train

tf.version

tf.xla

tf / recompute_grad

View aliases

Args

Returns