Searched refs:ThenMemcpyDeviceToDevice (Results 1 – 4 of 4) sorted by relevance
54 Status ThenMemcpyDeviceToDevice(se::Stream* transfer_stream,66 Status TpuDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::__anon240436b40111::TpuDeviceState
128 virtual Status ThenMemcpyDeviceToDevice(se::Stream* transfer_stream,
89 Status LocalDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::LocalDeviceState
1322 TF_RETURN_IF_ERROR(transfer_local_device->ThenMemcpyDeviceToDevice( in CopyToDeviceHelper()