Commit b300c78a authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

Fix while_v2 for XLA. 

Set max_num_elements on EmptyTensorList and add checks to make sure maximum_iterations is statically known in XLA context.
In order to make maximum_iterations available for the gradient pass it is set as an attr on the forward While op.

PiperOrigin-RevId: 220749927
parent cb576882
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