Searched refs:SynchronousMemcpyH2D (Results 1 – 11 of 11) 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() 179 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in AllocateRaw() 195 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in DeallocateRaw()
|
D | gpu_virtual_mem_allocator_test.cc | 73 ASSERT_TRUE(GpuDriver::SynchronousMemcpyH2D(gpu_context, gpu_buf, host_mem[0], in TEST()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream_executor_pimpl.h | 260 port::Status SynchronousMemcpyH2D(const void *host_src, int64 size, 267 port::Status SynchronousMemcpyH2D(port::ArraySlice<T> host_src, in SynchronousMemcpyH2D() function 271 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
|
D | stream_executor_pimpl.cc | 678 port::Status StreamExecutor::SynchronousMemcpyH2D( in SynchronousMemcpyH2D() function in stream_executor::StreamExecutor 685 SCOPED_TRACE(TraceListener::SynchronousMemcpyH2D, &result, host_src, size, in SynchronousMemcpyH2D()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_transfer_manager.cc | 168 Status s = executor->SynchronousMemcpyH2D( in TransferBufferToInfeedInternal()
|
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | gpu_driver.h | 341 static port::Status SynchronousMemcpyH2D(GpuContext* context,
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 486 return GpuDriver::SynchronousMemcpyH2D(context_, AsROCmDevicePtr(gpu_dst), in SynchronousMemcpy()
|
D | rocm_driver.cc | 847 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D( in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 578 return GpuDriver::SynchronousMemcpyH2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpy()
|
D | cuda_driver.cc | 1143 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D(GpuContext* context, in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor_test.cc | 804 TF_ASSERT_OK(executor->SynchronousMemcpyH2D(&src_data, size, &device_dst)); in TEST_F()
|