Home
last modified time | relevance | path

Searched refs:gpu_src (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc100 const DeviceMemoryBase &gpu_src, uint64 size) { in Memcpy() argument
103 void *src_mem = const_cast<void *>(gpu_src.opaque()); in Memcpy()
121 const DeviceMemoryBase &gpu_src, in MemcpyDeviceToDevice() argument
124 void *src_mem = const_cast<void *>(gpu_src.opaque()); in MemcpyDeviceToDevice()
171 const DeviceMemoryBase &gpu_src, in SynchronousMemcpy() argument
173 memcpy(host_dst, gpu_src.opaque(), size); in SynchronousMemcpy()
178 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() argument
179 memcpy(gpu_dst->opaque(), gpu_src.opaque(), size); in SynchronousMemcpyDeviceToDevice()
Dhost_gpu_executor.h75 bool Memcpy(Stream *stream, void *host_dst, const DeviceMemoryBase &gpu_src,
80 const DeviceMemoryBase &gpu_src,
101 const DeviceMemoryBase &gpu_src,
104 const DeviceMemoryBase &gpu_src,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc835 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, uint64 size) { in SynchronousMemcpyD2H() argument
838 tensorflow::wrap::hipMemcpyDtoH(host_dst, gpu_src, size), in SynchronousMemcpyD2H()
841 host_dst, absl::bit_cast<void*>(gpu_src), size, size)); in SynchronousMemcpyD2H()
863 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D() argument
867 tensorflow::wrap::hipMemcpyDtoD(gpu_dst, gpu_src, size), in SynchronousMemcpyD2D()
871 absl::bit_cast<void*>(gpu_dst), absl::bit_cast<void*>(gpu_src), size, in SynchronousMemcpyD2D()
879 hipDeviceptr_t gpu_src, in AsynchronousMemcpyD2H() argument
884 tensorflow::wrap::hipMemcpyDtoHAsync(host_dst, gpu_src, size, stream); in AsynchronousMemcpyD2H()
889 ToString(res).c_str(), host_dst, absl::bit_cast<void*>(gpu_src), size, in AsynchronousMemcpyD2H()
894 << " bytes from " << absl::bit_cast<void*>(gpu_src) << " to " in AsynchronousMemcpyD2H()
[all …]
Drocm_gpu_executor.cc491 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy() argument
494 AsROCmDevicePtr(gpu_src), size); in SynchronousMemcpy()
498 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() argument
500 AsROCmDevicePtr(gpu_src), size); in SynchronousMemcpyDeviceToDevice()
536 const DeviceMemoryBase& gpu_src, uint64 size) { in Memcpy() argument
538 AsROCmDevicePtr(gpu_src), size, in Memcpy()
551 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice() argument
554 AsROCmDevicePtr(gpu_src), size, in MemcpyDeviceToDevice()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h226 const DeviceMemoryBase &gpu_src,
229 DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src,
240 const DeviceMemoryBase &gpu_src, uint64 size) = 0;
244 const DeviceMemoryBase &gpu_src,
Dtrace_listener.h61 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyD2HBegin() argument
Dstream.h1722 Stream &ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src,
1736 Stream &ThenMemcpyD2H(const DeviceMemory<T> &gpu_src, in ThenMemcpyD2H() argument
1739 CHECK(gpu_src.size() == 0 || host_size >= gpu_src.size()); in ThenMemcpyD2H()
1740 return ThenMemcpy(host_dst.begin(), gpu_src, host_size); in ThenMemcpyD2H()
1757 Stream &ThenMemcpy(DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src,
1764 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpyD2D() argument
1765 return ThenMemcpy(gpu_dst, gpu_src, size); in ThenMemcpyD2D()
Dstream.cc4499 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy() argument
4501 VLOG_CALL(PARAM(host_dst), PARAM(gpu_src), PARAM(size)); in ThenMemcpy()
4503 CheckError(parent_->Memcpy(this, host_dst, gpu_src, size)); in ThenMemcpy()
4516 const DeviceMemoryBase &gpu_src, uint64 size) { in ThenMemcpy() argument
4517 VLOG_CALL(PARAM(gpu_dst), PARAM(gpu_src), PARAM(size)); in ThenMemcpy()
4519 CheckError(parent_->MemcpyDeviceToDevice(this, gpu_dst, gpu_src, size)); in ThenMemcpy()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h132 const DeviceMemoryBase& gpu_src,
136 const DeviceMemoryBase& gpu_src,
146 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src,
153 const DeviceMemoryBase& gpu_src,
Dgpu_driver.h340 GpuDevicePtr gpu_src, uint64 size);
346 GpuDevicePtr gpu_src, uint64 size);
352 GpuDevicePtr gpu_src, uint64 size,
358 GpuDevicePtr gpu_src, uint64 size,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc1130 CUdeviceptr gpu_src, in SynchronousMemcpyD2H() argument
1134 cuMemcpyDtoH(host_dst, gpu_src, size), in SynchronousMemcpyD2H()
1137 host_dst, absl::bit_cast<void*>(gpu_src), size, size)); in SynchronousMemcpyD2H()
1160 CUdeviceptr gpu_src, in SynchronousMemcpyD2D() argument
1164 cuMemcpyDtoD(gpu_dst, gpu_src, size), in SynchronousMemcpyD2D()
1168 absl::bit_cast<void*>(gpu_dst), absl::bit_cast<void*>(gpu_src), size, in SynchronousMemcpyD2D()
1176 CUdeviceptr gpu_src, in AsynchronousMemcpyD2H() argument
1180 CUresult res = cuMemcpyDtoHAsync(host_dst, gpu_src, size, stream); in AsynchronousMemcpyD2H()
1185 ToString(res), host_dst, absl::bit_cast<void*>(gpu_src), size, size); in AsynchronousMemcpyD2H()
1189 << " bytes from " << absl::bit_cast<void*>(gpu_src) << " to " in AsynchronousMemcpyD2H()
[all …]
Dcuda_gpu_executor.cc583 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy() argument
586 AsCudaDevicePtr(gpu_src), size); in SynchronousMemcpy()
590 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64 size) { in SynchronousMemcpyDeviceToDevice() argument
592 AsCudaDevicePtr(gpu_src), size); in SynchronousMemcpyDeviceToDevice()
628 const DeviceMemoryBase& gpu_src, uint64 size) { in Memcpy() argument
630 AsCudaDevicePtr(gpu_src), size, in Memcpy()
643 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice() argument
646 AsCudaDevicePtr(gpu_src), size, in MemcpyDeviceToDevice()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc362 const DeviceMemoryBase& gpu_src, in SynchronousMemcpy() argument
365 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(&gpu_src); in SynchronousMemcpy()
371 const DeviceMemoryBase& gpu_src, in SynchronousMemcpyDeviceToDevice() argument
375 SP_DeviceMemoryBase device_mem_src = DeviceMemoryBaseToC(&gpu_src); in SynchronousMemcpyDeviceToDevice()
395 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src, in Memcpy() argument
400 SP_DeviceMemoryBase device_mem_src = DeviceMemoryBaseToC(&gpu_src); in Memcpy()
424 const DeviceMemoryBase& gpu_src, in MemcpyDeviceToDevice() argument
430 SP_DeviceMemoryBase device_mem_src = DeviceMemoryBaseToC(&gpu_src); in MemcpyDeviceToDevice()
/external/tensorflow/tensorflow/
Dtensorflow.bzl1553 for gpu_src in gpu_srcs:
1554 if gpu_src.endswith(".cc") and not gpu_src.endswith(".cu.cc"):
1556 .format(gpu_src))