Searched refs:SynchronousMemcpyH2D (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 67 Status result = exec->SynchronousMemcpyH2D(mask, MASK_BYTES, &gpu_ptr); in InitMask()
177 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in AllocateRaw()
193 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in DeallocateRaw()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 272 port::Status SynchronousMemcpyH2D(const void *host_src, int64 size,
279 port::Status SynchronousMemcpyH2D(port::ArraySlice<T> host_src, in SynchronousMemcpyH2D() function
283 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
|
| D | stream_executor_pimpl.cc | 691 port::Status StreamExecutor::SynchronousMemcpyH2D( in SynchronousMemcpyH2D() function in stream_executor::StreamExecutor
698 SCOPED_TRACE(TraceListener::SynchronousMemcpyH2D, &result, host_src, size, in SynchronousMemcpyH2D()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 282 static port::Status SynchronousMemcpyH2D(GpuContext* context,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 167 Status s = executor->SynchronousMemcpyH2D( in TransferBufferToInfeedInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 225 TF_RETURN_IF_ERROR(executor->SynchronousMemcpyH2D( in ResolveConstantGlobals()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 487 return GpuDriver::SynchronousMemcpyH2D(context_, AsROCmDevicePtr(gpu_dst), in SynchronousMemcpy()
|
| D | rocm_driver.cc | 907 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D( in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 606 return GpuDriver::SynchronousMemcpyH2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpy()
|
| D | cuda_driver.cc | 1102 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D(GpuContext* context, in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
|