Searched defs:host_dst (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 51 bool XlaInterpreterExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
77 void *host_dst, const DeviceMemoryBase &dev_src, uint64 size) { in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_stream.h | 56 stream_executor::DeviceMemoryBase device_src, void* host_dst, in EnqueueTransferDeviceToHost()
|
| D | tpu_executor.cc | 279 bool TpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
308 void* host_dst, const ::stream_executor::DeviceMemoryBase& device_src, in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | trace_listener.h | 62 int64 size, void* host_dst) {} in SynchronousMemcpyD2HBegin()
|
| D | stream.h | 709 port::MutableArraySlice<ElementType> host_dst) { in ThenMemcpyD2HQuantized()
1737 port::MutableArraySlice<T> host_dst) { in ThenMemcpyD2H()
|
| D | stream_executor_pimpl.h | 283 port::MutableArraySlice<T> host_dst) { in SynchronousMemcpyD2H()
|
| D | stream_executor_pimpl.cc | 625 bool StreamExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy()
656 const DeviceMemoryBase &device_src, int64 size, void *host_dst) { in SynchronousMemcpyD2H()
701 bool StreamExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
|
| D | stream.cc | 1262 dnn::QuantizedActivationMode mode, void *host_dst, uint64 size) { in ThenMemcpyD2HQuantized()
4499 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 99 bool HostExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
170 port::Status HostExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 361 port::Status SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
395 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src, in Memcpy()
|
| D | stream_executor_test.cc | 95 void* host_dst, const SP_DeviceMemoryBase* const device_src, in memcpy_dtoh()
100 void sync_memcpy_dtoh(const SP_Device* const device, void* host_dst, in sync_memcpy_dtoh()
715 uint64_t size, TF_Status* const status) { in TEST_F()
777 uint64_t size, TF_Status* const status) { in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 490 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
535 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | rocm_driver.cc | 835 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, uint64 size) { in SynchronousMemcpyD2H()
878 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | rocm_dnn.cc | 4570 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 582 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
627 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | cuda_driver.cc | 1129 void* host_dst, in SynchronousMemcpyD2H()
1175 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | cuda_dnn.cc | 4609 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
|