Searched defs:gpu_dst (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 76 bool HostExecutor::Memcpy(Stream *stream, DeviceMemoryBase *gpu_dst, in Memcpy()
87 DeviceMemoryBase *gpu_dst, in MemcpyDeviceToDevice()
130 port::Status HostExecutor::SynchronousMemcpy(DeviceMemoryBase *gpu_dst, in SynchronousMemcpy()
145 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1706 DeviceMemory<T> *gpu_dst) { in ThenMemcpyH2D()
1721 Stream &ThenMemcpyD2D(DeviceMemoryBase *gpu_dst, in ThenMemcpyD2D()
|
| D | trace_listener.h | 56 DeviceMemoryBase* gpu_dst) {} in SynchronousMemcpyH2DBegin()
|
| D | stream.cc | 4793 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy()
4806 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, in ThenMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 485 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
499 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
543 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy()
551 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
| D | rocm_driver.cc | 908 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D()
925 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D()
963 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyH2D()
984 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyD2D()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 604 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
618 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
662 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy()
670 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
| D | cuda_driver.cc | 1103 CUdeviceptr gpu_dst, in SynchronousMemcpyH2D()
1126 CUdeviceptr gpu_dst, in SynchronousMemcpyD2D()
1176 CUdeviceptr gpu_dst, in AsynchronousMemcpyH2D()
1202 CUdeviceptr gpu_dst, in AsynchronousMemcpyD2D()
|