Commit a6c49167 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Allow gradients() calls from inside a tfe.defun wrt captured tensors. 

This modifies
https://github.com/tensorflow/tensorflow/commit/834da2c3fddab1bbbce742db572cfe65dd320fcd
to work with tfe.defun in addition to the legacy Defun implementation.

PiperOrigin-RevId: 211663702
parent 6aa8abbb
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