Searched refs:gpu_src_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| 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()
281 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local
283 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 420 se::DeviceMemoryBase gpu_src_ptr(DMAHelper::base(ctx->mutable_output(0)), in DoAddChain() local
423 gpu_src_ptr, return_bytes); in DoAddChain()
|