Searched refs:gpu_src_ptr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 161 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in SetProtoFromGPU() local
162 send_device_to_host_stream->ThenMemcpy(buf, gpu_src_ptr, total_bytes); in SetProtoFromGPU()
215 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in DeviceToDeviceCopy() local
232 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()
282 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local
284 send_device_to_host_stream->ThenMemcpy(dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToCPU()
440 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToSameGPU() local
443 send_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToSameGPU()
|