Skip to content

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

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

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