Searched defs:host_src (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 61 const void *host_src, uint64 size) { in Memcpy()
70 DeviceMemoryBase *dev_dst, const void *host_src, uint64 size) { in SynchronousMemcpy()
|
| D | executor.h | 88 const DeviceMemoryBase &host_src, in MemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 701 Stream &ThenMemcpyH2DQuantized(port::ArraySlice<ElementType> host_src, in ThenMemcpyH2DQuantized()
1705 Stream &ThenMemcpyH2D(port::ArraySlice<T> host_src, in ThenMemcpyH2D()
|
| D | trace_listener.h | 55 const void* host_src, int64 size, in SynchronousMemcpyH2DBegin()
|
| D | stream_executor_pimpl.cc | 622 const void *host_src, uint64 size) { in SynchronousMemcpy()
692 const void *host_src, int64 size, DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D()
720 const void *host_src, uint64 size) { in Memcpy()
|
| D | stream_executor_pimpl.h | 279 port::Status SynchronousMemcpyH2D(port::ArraySlice<T> host_src, in SynchronousMemcpyH2D()
|
| D | stream.cc | 1722 const void *host_src, uint64 size, dnn::QuantizedActivationMode mode, in ThenMemcpyH2DQuantized()
4793 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 77 const void *host_src, uint64 size) { in Memcpy()
131 const void *host_src, in SynchronousMemcpy()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 486 const void* host_src, uint64 size) { in SynchronousMemcpy()
544 const void* host_src, uint64 size) { in Memcpy()
|
| D | rocm_driver.cc | 908 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D()
964 const void* host_src, in AsynchronousMemcpyH2D()
|
| D | rocm_dnn.cc | 3979 Stream* stream, const void* host_src, int64 size, in DoMemcpyH2DQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 605 const void* host_src, uint64 size) { in SynchronousMemcpy()
663 const void* host_src, uint64 size) { in Memcpy()
|
| D | cuda_driver.cc | 1104 const void* host_src, in SynchronousMemcpyH2D()
1177 const void* host_src, in AsynchronousMemcpyH2D()
|
| D | cuda_dnn.cc | 4082 Stream* stream, const void* host_src, int64 size, in DoMemcpyH2DQuantized()
|