Searched defs:gpu_src (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.cc | 100 const DeviceMemoryBase &gpu_src, uint64 size) { in Memcpy() 121 const DeviceMemoryBase &gpu_src, in MemcpyDeviceToDevice() 171 const DeviceMemoryBase &gpu_src, in SynchronousMemcpy() 178 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | trace_listener.h | 61 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyD2HBegin()
|
D | stream.h | 1736 Stream &ThenMemcpyD2H(const DeviceMemory<T> &gpu_src, in ThenMemcpyD2H() 1764 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpyD2D()
|
D | stream.cc | 4499 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy() 4516 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpy()
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor.cc | 362 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy() 371 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyDeviceToDevice() 395 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src, in Memcpy() 424 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 583 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()
|
D | cuda_driver.cc | 1130 CUdeviceptr gpu_src, in SynchronousMemcpyD2H() 1160 CUdeviceptr gpu_src, in SynchronousMemcpyD2D() 1176 CUdeviceptr gpu_src, in AsynchronousMemcpyD2H() 1215 CUdeviceptr gpu_src, in AsynchronousMemcpyD2D()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 491 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy() 498 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() 536 const DeviceMemoryBase& gpu_src, uint64 size) { in Memcpy() 551 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
|
D | rocm_driver.cc | 835 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, uint64 size) { in SynchronousMemcpyD2H() 863 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D() 879 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2H() 922 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2D()
|