Home
last modified time | relevance | path

Searched defs:gpu_src (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc76 const DeviceMemoryBase &gpu_src, uint64 size) { in Memcpy()
97 const DeviceMemoryBase &gpu_src, in MemcpyDeviceToDevice()
147 const DeviceMemoryBase &gpu_src, in SynchronousMemcpy()
154 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1741 Stream &ThenMemcpyD2H(const DeviceMemory<T> &gpu_src, in ThenMemcpyD2H()
1769 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpyD2D()
Dtrace_listener.h61 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyD2HBegin()
Dstream.cc4922 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
4949 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpy()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc583 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy()
590 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
628 const DeviceMemoryBase& gpu_src, uint64 size) { in Memcpy()
643 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
Dcuda_driver.cc973 CUdeviceptr gpu_src, in SynchronousMemcpyD2H()
1003 CUdeviceptr gpu_src, in SynchronousMemcpyD2D()
1019 CUdeviceptr gpu_src, in AsynchronousMemcpyD2H()
1058 CUdeviceptr gpu_src, in AsynchronousMemcpyD2D()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc486 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy()
493 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
531 const DeviceMemoryBase& gpu_src, uint64 size) { in Memcpy()
546 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
Drocm_driver.cc828 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, uint64 size) { in SynchronousMemcpyD2H()
856 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D()
872 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2H()
915 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2D()