| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 95 void* host_dst, const SP_DeviceMemoryBase* const device_src, in memcpy_dtoh() argument
101 const SP_DeviceMemoryBase* const device_src, in sync_memcpy_dtoh() argument
714 const SP_DeviceMemoryBase* const device_src, in TEST_F()
718 std::memcpy(host_dst, device_src->opaque, size); in TEST_F()
727 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
728 Stream& stream_ref = stream.ThenMemcpy(&dst_data, device_src, size); in TEST_F()
756 const SP_DeviceMemoryBase* const device_src, in TEST_F()
759 std::memcpy(device_dst->opaque, device_src->opaque, size); in TEST_F()
769 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
770 stream.ThenMemcpy(&device_dst, device_src, size); in TEST_F()
[all …]
|
| D | stream_executor.h | 342 const SP_DeviceMemoryBase* device_src, uint64_t size,
355 const SP_DeviceMemoryBase* device_src, uint64_t size,
361 const SP_DeviceMemoryBase* device_src, uint64_t size,
375 const SP_DeviceMemoryBase* device_src, uint64_t size,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 626 const DeviceMemoryBase &device_src, in SynchronousMemcpy() argument
629 << ", device_src=" << device_src.opaque() << ", size=" << size in SynchronousMemcpy()
633 implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpy()
641 const DeviceMemoryBase &device_src, in SynchronousMemcpy() argument
644 << device_dst->opaque() << ", device_src=" << device_src.opaque() in SynchronousMemcpy()
648 device_dst, device_src, size); in SynchronousMemcpy()
656 const DeviceMemoryBase &device_src, int64 size, void *host_dst) { in SynchronousMemcpyD2H() argument
658 << device_src.opaque() << ", size=" << size in SynchronousMemcpyD2H()
662 SCOPED_TRACE(TraceListener::SynchronousMemcpyD2H, &result, device_src, size, in SynchronousMemcpyD2H()
665 result = implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpyD2H()
[all …]
|
| D | stream_executor_pimpl.h | 256 bool SynchronousMemcpy(void *host_dst, const DeviceMemoryBase &device_src,
275 port::Status SynchronousMemcpyD2H(const DeviceMemoryBase &device_src,
282 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H() argument
285 CHECK(device_src.size() == 0 || host_size >= device_src.size()); in SynchronousMemcpyD2H()
286 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
292 const DeviceMemoryBase &device_src,
573 const DeviceMemoryBase &device_src, uint64 size);
585 const DeviceMemoryBase &device_src, uint64 size);
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_stream.h | 56 stream_executor::DeviceMemoryBase device_src, void* host_dst, in EnqueueTransferDeviceToHost() argument
60 stream_, ApiConverter::ToC(device_src), host_dst, size, in EnqueueTransferDeviceToHost()
|
| D | tpu_executor.cc | 280 const ::stream_executor::DeviceMemoryBase& device_src, in Memcpy()
282 SE_DeviceMemoryBase se_base = ApiConverter::ToC(device_src); in Memcpy()
308 void* host_dst, const ::stream_executor::DeviceMemoryBase& device_src, in SynchronousMemcpy()
311 SE_DeviceMemoryBase se_base = ApiConverter::ToC(device_src); in SynchronousMemcpy()
319 const ::stream_executor::DeviceMemoryBase& device_src, uint64 size) { in SynchronousMemcpyDeviceToDevice()
|
| D | tpu_executor.h | 121 const ::stream_executor::DeviceMemoryBase& device_src,
138 void* host_dst, const ::stream_executor::DeviceMemoryBase& device_src,
142 const ::stream_executor::DeviceMemoryBase& device_src,
|
| D | tpu_executor_c_api.h | 90 const SE_DeviceMemoryBase* device_src,
98 const SE_DeviceMemoryBase* device_src,
135 SE_DeviceMemoryBase device_src,
|