Searched refs:CopyToDeviceHelper (Results 1 – 2 of 2) sorted by relevance
582 CopyToDeviceHelper(PjRtDevice* dst_device, LocalDeviceState* dst_local_device,
1293 PjRtStreamExecutorBuffer::CopyToDeviceHelper( in CopyToDeviceHelper() function in xla::PjRtStreamExecutorBuffer1404 buffer_and_event_or = CopyToDeviceHelper( in CopyToDevice()