Commit 9392ffa0 authored by Gaurav Jain's avatar Gaurav Jain Committed by TensorFlower Gardener
Browse files

Improve compatibility of while_v2 with XLA tests 

Remove assumption where resource variables could not be included as
outputs of the body. We instead iterate through the outputs to find
the first resource variable index.
Also loosen the requirements to specify maximum_iterations for XLA.

PiperOrigin-RevId: 226932912
parent 9585116b
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