Commit 62222398 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Update recompute_grad to work for functions without variables. 

Wrap recompute inputs in identity to avoid exponential graph traversal in gradients.

PiperOrigin-RevId: 203776272
parent deee57e9
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment