Home
last modified time | relevance | path

Searched refs:SynchronousMemcpyH2D (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc67 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()
Dgpu_virtual_mem_allocator_test.cc73 ASSERT_TRUE(GpuDriver::SynchronousMemcpyH2D(gpu_context, gpu_buf, host_mem[0], in TEST()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h260 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()
Dstream_executor_pimpl.cc678 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/
Dcpu_transfer_manager.cc168 Status s = executor->SynchronousMemcpyH2D( in TransferBufferToInfeedInternal()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h341 static port::Status SynchronousMemcpyH2D(GpuContext* context,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc486 return GpuDriver::SynchronousMemcpyH2D(context_, AsROCmDevicePtr(gpu_dst), in SynchronousMemcpy()
Drocm_driver.cc847 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D( in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc578 return GpuDriver::SynchronousMemcpyH2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpy()
Dcuda_driver.cc1143 /* static */ port::Status GpuDriver::SynchronousMemcpyH2D(GpuContext* context, in SynchronousMemcpyH2D() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc804 TF_ASSERT_OK(executor->SynchronousMemcpyH2D(&src_data, size, &device_dst)); in TEST_F()