| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 51 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&gpu_array_ptr, &cpu_array[0], in TEST()
80 ASSERT_TRUE(stream_exec->SynchronousMemcpy( in TEST()
88 stream_exec->SynchronousMemcpy(&gpu_hdr_ptr, &pi, sizeof(float))); in TEST()
116 ASSERT_TRUE(stream_exec->SynchronousMemcpy( in TEST()
124 stream_exec->SynchronousMemcpy(&gpu_ftr_ptr, &pi, sizeof(float))); in TEST()
149 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&cpu_array[0], gpu_array_ptr, in TEST()
157 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&gpu_array_ptr, &cpu_array[0], in TEST()
161 stream_exec->SynchronousMemcpy(&cpu_array_result[0], gpu_array_ptr, in TEST()
170 stream_exec->SynchronousMemcpy(&cpu_array_result[0], gpu_array_ptr, in TEST()
196 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&cpu_array[0], gpu_array_ptr, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 54 port::Status ok = SynchronousMemcpy(host_dst, dev_src, size); in Memcpy()
63 port::Status ok = SynchronousMemcpy(dev_dst, host_src, size); in Memcpy()
69 port::Status XlaInterpreterExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in stream_executor::interpreter::XlaInterpreterExecutor
75 port::Status XlaInterpreterExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 117 port::Status SynchronousMemcpy(DeviceMemoryBase *pop_dst,
119 port::Status SynchronousMemcpy(void *host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 621 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
631 implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpy()
638 bool StreamExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
646 implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpy()
653 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
678 result = implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpyD2H()
701 result = implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpyH2D()
|
| D | stream_executor_pimpl.h | 261 bool SynchronousMemcpy(DeviceMemoryBase *device_dst, const void *host_src,
268 bool SynchronousMemcpy(void *host_dst, const DeviceMemoryBase &device_src,
303 bool SynchronousMemcpy(DeviceMemoryBase *device_dst,
796 if (!parent->SynchronousMemcpy(ptr(), const_cast<const ElemT *>(&local[0]), in ScopedDeviceMemory()
|
| D | stream_executor_internal.h | 219 virtual port::Status SynchronousMemcpy(DeviceMemoryBase *gpu_dst,
221 virtual port::Status SynchronousMemcpy(void *host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 97 port::Status SynchronousMemcpy(DeviceMemoryBase *gpu_dst,
99 port::Status SynchronousMemcpy(void *host_dst,
|
| D | host_gpu_executor.cc | 130 port::Status HostExecutor::SynchronousMemcpy(DeviceMemoryBase *gpu_dst, in SynchronousMemcpy() function in stream_executor::host::HostExecutor
137 port::Status HostExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 127 port::Status SynchronousMemcpy(DeviceMemoryBase* gpu_dst,
130 port::Status SynchronousMemcpy(void* host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 485 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
491 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 604 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
610 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
|