xla

[while_loop] prepare for torch while_loop signature change.

#6872

Merged