Searched defs:host_dst (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 50 bool XlaInterpreterExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
76 void *host_dst, const DeviceMemoryBase &dev_src, uint64 size) { in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 686 port::MutableArraySlice<ElementType> host_dst) { in ThenMemcpyD2HQuantized()
1695 port::MutableArraySlice<T> host_dst) { in ThenMemcpyD2H()
|
| D | trace_listener.h | 62 int64 size, void* host_dst) {} in SynchronousMemcpyD2HBegin()
|
| D | stream_executor_pimpl.cc | 638 bool StreamExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy()
669 const DeviceMemoryBase &device_src, int64 size, void *host_dst) { in SynchronousMemcpyD2H()
714 bool StreamExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
|
| D | stream_executor_pimpl.h | 295 port::MutableArraySlice<T> host_dst) { in SynchronousMemcpyD2H()
|
| D | stream.cc | 1706 dnn::QuantizedActivationMode mode, void *host_dst, uint64 size) { in ThenMemcpyD2HQuantized()
4780 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 66 bool HostExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
137 port::Status HostExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 491 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
536 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | rocm_driver.cc | 892 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, uint64 size) { in SynchronousMemcpyD2H()
941 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | rocm_dnn.cc | 3973 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 610 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
655 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | cuda_driver.cc | 1079 void* host_dst, in SynchronousMemcpyD2H()
1149 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | cuda_dnn.cc | 4076 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
|