Searched refs:SynchronousMemcpyD2H (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 275 port::Status SynchronousMemcpyD2H(const DeviceMemoryBase &device_src,
282 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H() function
286 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
|
| D | stream_executor_pimpl.cc | 655 port::Status StreamExecutor::SynchronousMemcpyD2H( in SynchronousMemcpyD2H() function in stream_executor::StreamExecutor
662 SCOPED_TRACE(TraceListener::SynchronousMemcpyD2H, &result, device_src, size, in SynchronousMemcpyD2H()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator_test.cc | 76 ASSERT_TRUE(GpuDriver::SynchronousMemcpyD2H(gpu_context, host_mem[1], gpu_buf, in TEST()
|
| D | gpu_debug_allocator.cc | 47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()
|
| D | gpu_util.cc | 384 auto s = dev_info->stream->parent()->SynchronousMemcpyD2H( in MemoryDebugString()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 339 static port::Status SynchronousMemcpyD2H(GpuContext* context, void* host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 493 return GpuDriver::SynchronousMemcpyD2H(context_, host_dst, in SynchronousMemcpy()
|
| D | rocm_driver.cc | 834 /* static */ port::Status GpuDriver::SynchronousMemcpyD2H( in SynchronousMemcpyD2H() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 585 return GpuDriver::SynchronousMemcpyD2H(context_, host_dst, in SynchronousMemcpy()
|
| D | cuda_driver.cc | 1128 /* static */ port::Status GpuDriver::SynchronousMemcpyD2H(GpuContext* context, in SynchronousMemcpyD2H() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 787 TF_ASSERT_OK(executor->SynchronousMemcpyD2H(device_src, size, &dst_data)); in TEST_F()
|