jax
61c76d45 - Merge pull request #2738 from chr1sj0nes/patch-1

Commit

5 years ago

Merge pull request #2738 from chr1sj0nes/patch-1 Actually use xla._device_put_scalar for device-putting scalars

References

Author

mattjj

Parents