Searched defs:gpu_src (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 70 const DeviceMemoryBase &gpu_src, uint64 size) { in Memcpy()
91 const DeviceMemoryBase &gpu_src, in MemcpyDeviceToDevice()
141 const DeviceMemoryBase &gpu_src, in SynchronousMemcpy()
148 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1711 Stream &ThenMemcpyD2H(const DeviceMemory<T> &gpu_src, in ThenMemcpyD2H()
1739 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpyD2D()
|
| D | trace_listener.h | 62 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyD2HBegin()
|
| D | stream.cc | 4617 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
4644 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 512 const DeviceMemoryBase &gpu_src, in SynchronousMemcpy()
519 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
557 const DeviceMemoryBase &gpu_src, uint64 size) { in Memcpy()
572 const DeviceMemoryBase &gpu_src, in MemcpyDeviceToDevice()
|
| D | cuda_driver.cc | 1151 CUdeviceptr gpu_src, in SynchronousMemcpyD2H()
1186 CUdeviceptr gpu_src, in SynchronousMemcpyD2D()
1203 CUdeviceptr gpu_src, in AsynchronousMemcpyD2H()
1242 CUdeviceptr gpu_src, in AsynchronousMemcpyD2D()
|