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

while_v2: handle resource tensors captured from function call output. 

We could still use more test coverage, such as calling a function and
then taking the gradient in a while loop, then taking the gradient of
the outer while loop. I ran into other errors writing this test case
though.

PiperOrigin-RevId: 232052854
parent 722510d6
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