Parallel-for: Change while_loop conversion so that any resource/variant tensors

directly entering the body of the original while_loop also enters directly
in the converted while_loop.

PiperOrigin-RevId: 203183248