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()
|