Searched refs:recv_host_to_device_stream (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_util.cc | 235 auto recv_host_to_device_stream = in CopyCPUTensorToPluggableDevice() local
238 if (recv_host_to_device_stream == nullptr) { in CopyCPUTensorToPluggableDevice()
245 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToPluggableDevice()
253 recv_host_to_device_stream->ThenMemcpy(&device_dst_ptr, src_ptr, in CopyCPUTensorToPluggableDevice()
259 recv_host_to_device_stream, in CopyCPUTensorToPluggableDevice()
260 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToPluggableDevice()
262 if (!recv_host_to_device_stream->ok()) { in CopyCPUTensorToPluggableDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 313 auto recv_host_to_device_stream = in CopyCPUTensorToGPU() local
316 if (recv_host_to_device_stream == nullptr) { in CopyCPUTensorToGPU()
322 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()
331 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, total_bytes); in CopyCPUTensorToGPU()
336 recv_host_to_device_stream, in CopyCPUTensorToGPU()
337 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToGPU()
339 if (!recv_host_to_device_stream->ok()) { in CopyCPUTensorToGPU()
|