| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 98 SP_DeviceMemoryBase* const device_dst, const void* host_src, in memcpy_htod() argument
104 SP_DeviceMemoryBase* const device_dst, in sync_memcpy_htod() argument
735 SP_DeviceMemoryBase* const device_dst, in TEST_F()
739 std::memcpy(device_dst->opaque, host_src, size); in TEST_F()
748 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
749 stream.ThenMemcpy(&device_dst, &src_data, size); in TEST_F()
755 SP_DeviceMemoryBase* const device_dst, in TEST_F() argument
759 std::memcpy(device_dst->opaque, device_src->opaque, size); in TEST_F()
768 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
770 stream.ThenMemcpy(&device_dst, device_src, size); in TEST_F()
[all …]
|
| D | stream_executor.h | 348 SP_DeviceMemoryBase* device_dst, const void* host_src,
354 SP_DeviceMemoryBase* device_dst,
367 SP_DeviceMemoryBase* device_dst,
374 SP_DeviceMemoryBase* device_dst,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 608 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() argument
611 << device_dst->opaque() << ", host_src=" << host_src in SynchronousMemcpy()
618 implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpy()
640 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() argument
644 << device_dst->opaque() << ", device_src=" << device_src.opaque() in SynchronousMemcpy()
648 device_dst, device_src, size); in SynchronousMemcpy()
679 const void *host_src, int64 size, DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D() argument
681 << ", size=" << size << ", device_dst=" << device_dst->opaque() << ")" in SynchronousMemcpyH2D()
686 device_dst); in SynchronousMemcpyH2D()
688 result = implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpyH2D()
[all …]
|
| D | stream_executor_pimpl.h | 249 bool SynchronousMemcpy(DeviceMemoryBase *device_dst, const void *host_src,
261 DeviceMemoryBase *device_dst);
268 DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D() argument
270 CHECK(device_dst->size() == 0 || device_dst->size() >= host_size); in SynchronousMemcpyH2D()
271 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
291 bool SynchronousMemcpy(DeviceMemoryBase *device_dst,
577 bool Memcpy(Stream *stream, DeviceMemoryBase *device_dst,
584 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *device_dst,
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_stream.h | 46 stream_executor::DeviceMemoryBase device_dst, const void* host_src, in EnqueueTransferHostToDevice() argument
50 stream_, ApiConverter::ToC(device_dst), const_cast<void*>(host_src), in EnqueueTransferHostToDevice()
|
| D | tpu_executor.cc | 289 ::stream_executor::DeviceMemoryBase* device_dst, in Memcpy() argument
291 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*device_dst); in Memcpy()
298 ::stream_executor::DeviceMemoryBase* device_dst, const void* host_src, in SynchronousMemcpy() argument
301 SE_DeviceMemoryBase se_base = ApiConverter::ToC(*device_dst); in SynchronousMemcpy()
318 ::stream_executor::DeviceMemoryBase* device_dst, in SynchronousMemcpyDeviceToDevice() argument
|
| D | tpu_executor.h | 124 bool Memcpy(Stream* stream, ::stream_executor::DeviceMemoryBase* device_dst,
135 Status SynchronousMemcpy(::stream_executor::DeviceMemoryBase* device_dst,
141 ::stream_executor::DeviceMemoryBase* device_dst,
|
| D | tpu_executor_c_api.h | 93 SE_DeviceMemoryBase* device_dst,
102 SE_DeviceMemoryBase* device_dst,
131 SE_DeviceMemoryBase device_dst,
|