Searched refs:device_dst (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 621 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() argument
624 << device_dst->opaque() << ", host_src=" << host_src in SynchronousMemcpy()
631 implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpy()
653 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() argument
657 << device_dst->opaque() << ", device_src=" << device_src.opaque() in SynchronousMemcpy()
661 device_dst, device_src, size); in SynchronousMemcpy()
692 const void *host_src, int64 size, DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D() argument
694 << ", size=" << size << ", device_dst=" << device_dst->opaque() << ")" in SynchronousMemcpyH2D()
699 device_dst); in SynchronousMemcpyH2D()
701 result = implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpyH2D()
[all …]
|
| D | stream_executor_pimpl.h | 261 bool SynchronousMemcpy(DeviceMemoryBase *device_dst, const void *host_src,
273 DeviceMemoryBase *device_dst);
280 DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D() argument
282 CHECK(device_dst->size() == 0 || device_dst->size() >= host_size); in SynchronousMemcpyH2D()
283 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
303 bool SynchronousMemcpy(DeviceMemoryBase *device_dst,
551 bool Memcpy(Stream *stream, DeviceMemoryBase *device_dst,
558 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *device_dst,
|