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

Don't constant fold LoopCond nodes. 

Removing the LoopCond of a while_loop can cause the partitioner to fail with:
  A cross-device loop must have a pivot predicate

For some reason this only triggers with while_v2 (the lowered while
loop is slightly different than what would be produced by the original
while_loop).

PiperOrigin-RevId: 225188075
parent bf16a751
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