xla
xla copied to clipboard
Add a version of CreateBuffersForAsyncHostToDevice that takes a custom layout.
Add a version of CreateBuffersForAsyncHostToDevice that takes a custom layout.