Home
last modified time | relevance | path

Searched refs:device_src (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc639 const DeviceMemoryBase &device_src, in SynchronousMemcpy() argument
642 << ", device_src=" << device_src.opaque() << ", size=" << size in SynchronousMemcpy()
646 implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpy()
654 const DeviceMemoryBase &device_src, in SynchronousMemcpy() argument
657 << device_dst->opaque() << ", device_src=" << device_src.opaque() in SynchronousMemcpy()
661 device_dst, device_src, size); in SynchronousMemcpy()
669 const DeviceMemoryBase &device_src, int64 size, void *host_dst) { in SynchronousMemcpyD2H() argument
671 << device_src.opaque() << ", size=" << size in SynchronousMemcpyD2H()
675 SCOPED_TRACE(TraceListener::SynchronousMemcpyD2H, &result, device_src, size, in SynchronousMemcpyD2H()
678 result = implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpyD2H()
[all …]
Dstream_executor_pimpl.h268 bool SynchronousMemcpy(void *host_dst, const DeviceMemoryBase &device_src,
287 port::Status SynchronousMemcpyD2H(const DeviceMemoryBase &device_src,
294 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H() argument
297 CHECK(device_src.size() == 0 || host_size >= device_src.size()); in SynchronousMemcpyD2H()
298 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
304 const DeviceMemoryBase &device_src,
547 const DeviceMemoryBase &device_src, uint64 size);
559 const DeviceMemoryBase &device_src, uint64 size);