Searched refs:recv_host_to_device_stream (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 325 auto recv_host_to_device_stream = in CopyCPUTensorToGPU() local
328 if (recv_host_to_device_stream == nullptr) { in CopyCPUTensorToGPU()
334 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()
368 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, staging_buffer, in CopyCPUTensorToGPU()
371 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, in CopyCPUTensorToGPU()
377 recv_host_to_device_stream, in CopyCPUTensorToGPU()
378 [recv_host_to_device_stream, done, input_ref, do_staging, staging_buffer, in CopyCPUTensorToGPU()
385 if (!recv_host_to_device_stream->ok()) { in CopyCPUTensorToGPU()
|