Searched refs:SynchronousMemcpyD2H (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 287 port::Status SynchronousMemcpyD2H(const DeviceMemoryBase &device_src,
294 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H() function
298 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
|
| D | stream_executor_pimpl.cc | 668 port::Status StreamExecutor::SynchronousMemcpyD2H( in SynchronousMemcpyD2H() function in stream_executor::StreamExecutor
675 SCOPED_TRACE(TraceListener::SynchronousMemcpyD2H, &result, device_src, size, in SynchronousMemcpyD2H()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 87 TF_RETURN_IF_ERROR(executor->SynchronousMemcpyD2H( in TransferLiteralFromDeviceInternal()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()
|
| D | gpu_util.cc | 383 auto s = dev_info->stream->parent()->SynchronousMemcpyD2H( in MemoryDebugString()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 280 static port::Status SynchronousMemcpyD2H(GpuContext* context, void* host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 494 return GpuDriver::SynchronousMemcpyD2H(context_, host_dst, in SynchronousMemcpy()
|
| D | rocm_driver.cc | 891 /* static */ port::Status GpuDriver::SynchronousMemcpyD2H( in SynchronousMemcpyD2H() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 613 return GpuDriver::SynchronousMemcpyD2H(context_, host_dst, in SynchronousMemcpy()
|
| D | cuda_driver.cc | 1078 /* static */ port::Status GpuDriver::SynchronousMemcpyD2H(GpuContext* context, in SynchronousMemcpyD2H() function in stream_executor::gpu::GpuDriver
|