xla
f77d1fac - [Backport] Move CUDA tensors to CPU before moving to XLA (#6066)

Commit

1 year ago

[Backport] Move CUDA tensors to CPU before moving to XLA (#6066) Co-authored-by: Yukio Siraichi <yukio.siraichi@gmail.com>

References

Author

will-cromar

Parents