Searched refs:send_device_to_host_stream (Results 1 – 1 of 1) sorted by relevance
127 auto send_device_to_host_stream = in SetProtoFromGPU() local130 if (send_device_to_host_stream == nullptr) { in SetProtoFromGPU()135 send_device_to_host_stream->ThenWaitFor(send_stream); in SetProtoFromGPU()161 send_device_to_host_stream->ThenMemcpy(buf, gpu_src_ptr, total_bytes); in SetProtoFromGPU()166 send_device_to_host_stream, [send_device_to_host_stream, done, proto, buf, in SetProtoFromGPU()168 if (!send_device_to_host_stream->ok()) { in SetProtoFromGPU()268 auto send_device_to_host_stream = in CopyGPUTensorToCPU() local271 if (send_device_to_host_stream == nullptr) { in CopyGPUTensorToCPU()276 send_device_to_host_stream->ThenWaitFor(send_stream); in CopyGPUTensorToCPU()283 send_device_to_host_stream->ThenMemcpy(dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToCPU()[all …]