xla
xla copied to clipboard
[PJRT:C] Add CopyToDeviceStream to PJRT C API.
[PJRT:C] Add CopyToDeviceStream to PJRT C API.
This is part of what's needed to make send/recv host callbacks work.