xla

[Backport] Move CUDA tensors to CPU before moving to XLA

#6066

Merged