Searched defs:host_src (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | executor.cc | 62 const void *host_src, uint64 size) { in Memcpy() 71 DeviceMemoryBase *dev_dst, const void *host_src, uint64 size) { in SynchronousMemcpy()
|
D | executor.h | 87 const DeviceMemoryBase &host_src, in MemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/tpu/ |
D | tpu_stream.h | 46 stream_executor::DeviceMemoryBase device_dst, const void* host_src, in EnqueueTransferHostToDevice()
|
D | tpu_executor.cc | 290 const void* host_src, uint64 size) { in Memcpy() 298 ::stream_executor::DeviceMemoryBase* device_dst, const void* host_src, in SynchronousMemcpy() 324 bool TpuExecutor::MemcpyDeviceToDevice( in MemcpyDeviceToDevice()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | trace_listener.h | 55 const void* host_src, int64 size, in SynchronousMemcpyH2DBegin()
|
D | stream.h | 724 Stream &ThenMemcpyH2DQuantized(port::ArraySlice<ElementType> host_src, in ThenMemcpyH2DQuantized() 1747 Stream &ThenMemcpyH2D(port::ArraySlice<T> host_src, in ThenMemcpyH2D()
|
D | stream_executor_pimpl.h | 267 port::Status SynchronousMemcpyH2D(port::ArraySlice<T> host_src, in SynchronousMemcpyH2D()
|
D | stream_executor_pimpl.cc | 609 const void *host_src, uint64 size) { in SynchronousMemcpy() 679 const void *host_src, int64 size, DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D() 707 const void *host_src, uint64 size) { in Memcpy()
|
D | stream.cc | 1276 const void *host_src, uint64 size, dnn::QuantizedActivationMode mode, in ThenMemcpyH2DQuantized() 4507 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy()
|
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.cc | 110 const void *host_src, uint64 size) { in Memcpy() 164 const void *host_src, in SynchronousMemcpy()
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor.cc | 354 const void* host_src, uint64 size) override { in SynchronousMemcpy() 409 bool Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, const void* host_src, in Memcpy()
|
D | stream_executor_test.cc | 98 SP_DeviceMemoryBase* const device_dst, const void* host_src, in memcpy_htod() 105 const void* host_src, uint64_t size, in sync_memcpy_htod() 737 TF_Status* const status) { in TEST_F() 794 const void* host_src, uint64_t size, TF_Status* const status) { in TEST_F()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 485 const void* host_src, uint64 size) { in SynchronousMemcpy() 543 const void* host_src, uint64 size) { in Memcpy()
|
D | rocm_driver.cc | 848 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D() 901 const void* host_src, in AsynchronousMemcpyH2D()
|
D | rocm_dnn.cc | 4576 Stream* stream, const void* host_src, int64 size, in DoMemcpyH2DQuantized()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 577 const void* host_src, uint64 size) { in SynchronousMemcpy() 635 const void* host_src, uint64 size) { in Memcpy()
|
D | cuda_driver.cc | 1145 const void* host_src, in SynchronousMemcpyH2D() 1196 const void* host_src, in AsynchronousMemcpyH2D()
|
D | cuda_dnn.cc | 4615 Stream* stream, const void* host_src, int64 size, in DoMemcpyH2DQuantized()
|