Searched refs:gpu_src_ptr (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 160 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in SetProtoFromGPU() local
161 send_device_to_host_stream->ThenMemcpy(buf, gpu_src_ptr, total_bytes); in SetProtoFromGPU()
214 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in DeviceToDeviceCopy() local
231 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()
293 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local
295 send_device_to_host_stream->ThenMemcpy(dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToCPU()
486 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToSameGPU() local
489 send_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToSameGPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_event_mgr_test.cc | 419 se::DeviceMemoryBase gpu_src_ptr(DMAHelper::base(ctx->mutable_output(0)), in DoAddChain() local
422 gpu_src_ptr, return_bytes); in DoAddChain()
|