Home
last modified time | relevance | path

Searched defs:gpu_dst (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc109 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/
Dtrace_listener.h56 DeviceMemoryBase* gpu_dst) {} in SynchronousMemcpyH2DBegin()
Dstream.h1748 DeviceMemory<T> *gpu_dst) { in ThenMemcpyH2D()
1763 Stream &ThenMemcpyD2D(DeviceMemoryBase *gpu_dst, in ThenMemcpyD2D()
Dstream.cc4507 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/
Dstream_executor.cc353 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/
Dcuda_gpu_executor.cc576 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()
Dcuda_driver.cc1144 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/
Drocm_gpu_executor.cc484 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()
Drocm_driver.cc848 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/
Dtpu_executor.cc325 Stream* stream, ::stream_executor::DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()