Commit bbde9a6c authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

Add a new argument `return_same_shape` to `while_loop`. When set, the... 

Add a new argument `return_same_shape` to `while_loop`. When set, the structure of the return value will be exactly the same as that of the loop_vars.

PiperOrigin-RevId: 203150128
parent 31a5fa1e
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