Searched defs:gpu_dst (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.cc | 109 bool HostExecutor::Memcpy(Stream *stream, DeviceMemoryBase *gpu_dst, in Memcpy() 120 DeviceMemoryBase *gpu_dst, in MemcpyDeviceToDevice() 163 port::Status HostExecutor::SynchronousMemcpy(DeviceMemoryBase *gpu_dst, in SynchronousMemcpy() 178 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | trace_listener.h | 56 DeviceMemoryBase* gpu_dst) {} in SynchronousMemcpyH2DBegin()
|
D | stream.h | 1748 DeviceMemory<T> *gpu_dst) { in ThenMemcpyH2D() 1763 Stream &ThenMemcpyD2D(DeviceMemoryBase *gpu_dst, in ThenMemcpyD2D()
|
D | stream.cc | 4507 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy() 4515 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, in ThenMemcpy()
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor.cc | 353 port::Status SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() 370 port::Status SynchronousMemcpyDeviceToDevice(DeviceMemoryBase* gpu_dst, in SynchronousMemcpyDeviceToDevice() 409 bool Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, const void* host_src, in Memcpy() 423 bool MemcpyDeviceToDevice(Stream* stream, DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 576 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() 590 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() 634 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy() 642 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
D | cuda_driver.cc | 1144 CUdeviceptr gpu_dst, in SynchronousMemcpyH2D() 1159 CUdeviceptr gpu_dst, in SynchronousMemcpyD2D() 1195 CUdeviceptr gpu_dst, in AsynchronousMemcpyH2D() 1214 CUdeviceptr gpu_dst, in AsynchronousMemcpyD2D()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 484 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() 498 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() 542 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy() 550 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
D | rocm_driver.cc | 848 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D() 863 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D() 900 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyH2D() 921 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyD2D()
|
/external/tensorflow/tensorflow/stream_executor/tpu/ |
D | tpu_executor.cc | 325 Stream* stream, ::stream_executor::DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|