Searched refs:dst_local_device (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1294 PjRtDevice* dst_device, LocalDeviceState* dst_local_device, in CopyToDeviceHelper() argument
1300 dst_device, dst_local_device, transfer_stream, in CopyToDeviceHelper()
1323 transfer_stream, dst_local_device->compute_stream(), input_buffer, in CopyToDeviceHelper()
1336 if (transfer_local_device == dst_local_device) { in CopyToDeviceHelper()
1375 LocalDeviceState * dst_local_device, in CopyToDevice()
1383 : dst_local_device; in CopyToDevice()
1384 CHECK_EQ(dst_local_device->allocation_model(), in CopyToDevice()
1405 dst_device, dst_local_device, transfer_local_device, transfer_stream, in CopyToDevice()
|
| D | pjrt_stream_executor_client.h | 582 CopyToDeviceHelper(PjRtDevice* dst_device, LocalDeviceState* dst_local_device,
|