Searched defs:gpu_src (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.cc | 100 const DeviceMemoryBase& gpu_src, uint64_t 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_t size) { in SynchronousMemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | trace_listener.h | 61 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyD2HBegin()
|
| D | stream.h | 1182 Stream &ThenMemcpyD2H(const DeviceMemory<T> &gpu_src, in ThenMemcpyD2H()
1210 const DeviceMemoryBase &gpu_src, uint64_t size) { in ThenMemcpyD2D()
|
| D | stream.cc | 1991 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
2008 const DeviceMemoryBase &gpu_src, uint64_t size) { in ThenMemcpy()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 334 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy()
343 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyDeviceToDevice()
382 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src, in Memcpy()
411 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 494 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy()
501 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64_t size) { in SynchronousMemcpyDeviceToDevice()
539 const DeviceMemoryBase& gpu_src, uint64_t size) { in Memcpy()
554 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
|
| D | rocm_driver.cc | 844 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2H()
873 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D()
889 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2H()
932 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2D()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 626 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy()
633 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64_t size) { in SynchronousMemcpyDeviceToDevice()
671 const DeviceMemoryBase& gpu_src, uint64_t size) { in Memcpy()
686 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice()
|
| D | cuda_driver.cc | 1075 CUdeviceptr gpu_src, in SynchronousMemcpyD2H()
1105 CUdeviceptr gpu_src, in SynchronousMemcpyD2D()
1151 CUdeviceptr gpu_src, in AsynchronousMemcpyD2H()
1190 CUdeviceptr gpu_src, in AsynchronousMemcpyD2D()
|