Searched defs:host_dst (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 51 bool XlaInterpreterExecutor::Memcpy(Stream *stream, void *host_dst, in Memcpy()
93 void *host_dst, const DeviceMemoryBase &dev_src, uint64_t size) { in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/compiler/xla/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/compiler/xla/stream_executor/ |
| D | trace_listener.h | 62 int64_t size, void* host_dst) {} in SynchronousMemcpyD2HBegin()
|
| D | stream_executor_pimpl.h | 286 port::MutableArraySlice<T> host_dst) { in SynchronousMemcpyD2H()
|
| D | stream_executor_pimpl.cc | 628 bool StreamExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
659 const DeviceMemoryBase& device_src, int64_t size, void* host_dst) { in SynchronousMemcpyD2H()
704 bool StreamExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | stream.h | 618 port::MutableArraySlice<ElementType> host_dst) { in ThenMemcpyD2HQuantized()
1183 port::MutableArraySlice<T> host_dst) { in ThenMemcpyD2H()
|
| D | stream.cc | 894 dnn::QuantizedActivationMode mode, void *host_dst, uint64_t size) { in ThenMemcpyD2HQuantized()
1991 Stream &Stream::ThenMemcpy(void *host_dst, const DeviceMemoryBase &gpu_src, in ThenMemcpy()
|
| /external/tensorflow/tensorflow/compiler/xla/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_test_util.cc | 67 void MemcpyDToH(const SP_Device* const device, SP_Stream stream, void* host_dst, in MemcpyDToH()
73 void SyncMemcpyDToH(const SP_Device* const device, void* host_dst, in SyncMemcpyDToH()
|
| D | stream_executor.cc | 333 port::Status SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
382 bool Memcpy(Stream* stream, void* host_dst, const DeviceMemoryBase& gpu_src, in Memcpy()
|
| D | stream_executor_test.cc | 539 uint64_t size, TF_Status* const status) { in TEST_F()
601 uint64_t size, TF_Status* const status) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | while_op_test.cc | 109 uint64_t size, TF_Status* status) { in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 493 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
538 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | rocm_driver.cc | 844 GpuContext* context, void* host_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2H()
888 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | rocm_dnn.cc | 4510 dnn::QuantizedActivationMode mode, void* host_dst, int64_t size) { in DoMemcpyD2HQuantized()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 625 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy()
670 bool GpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
|
| D | cuda_driver.cc | 1074 void* host_dst, in SynchronousMemcpyD2H()
1150 void* host_dst, in AsynchronousMemcpyD2H()
|
| D | cuda_dnn.cc | 6520 dnn::QuantizedActivationMode mode, void* host_dst, int64_t size) { in DoMemcpyD2HQuantized()
|