Commit 8e5c118c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

While loop dispatch depends only on whether variables directly referenced in... 

While loop dispatch depends only on whether variables directly referenced in the condition are tensors.

This fixes a bug where a variable in an inner loop could be referenced before creation. These variables would be used in the AG while_stmt to determine whether to dispatch to tf.while_loop or run the Python loop.

PiperOrigin-RevId: 210550604
parent 8987d1cf
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