| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 61 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&gpu_array_ptr, &cpu_array[0], in TEST()
90 ASSERT_TRUE(stream_exec->SynchronousMemcpy( in TEST()
98 stream_exec->SynchronousMemcpy(&gpu_hdr_ptr, &pi, sizeof(float))); in TEST()
126 ASSERT_TRUE(stream_exec->SynchronousMemcpy( in TEST()
134 stream_exec->SynchronousMemcpy(&gpu_ftr_ptr, &pi, sizeof(float))); in TEST()
157 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&cpu_array[0], gpu_array_ptr, in TEST()
165 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&gpu_array_ptr, &cpu_array[0], in TEST()
169 stream_exec->SynchronousMemcpy(&cpu_array_result[0], gpu_array_ptr, in TEST()
178 stream_exec->SynchronousMemcpy(&cpu_array_result[0], gpu_array_ptr, in TEST()
202 ASSERT_TRUE(stream_exec->SynchronousMemcpy(&cpu_array[0], gpu_array_ptr, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 55 port::Status ok = SynchronousMemcpy(host_dst, dev_src, size); in Memcpy()
64 port::Status ok = SynchronousMemcpy(dev_dst, host_src, size); in Memcpy()
70 port::Status XlaInterpreterExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in stream_executor::interpreter::XlaInterpreterExecutor
76 port::Status XlaInterpreterExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 117 port::Status SynchronousMemcpy(DeviceMemoryBase *dev_dst,
119 port::Status SynchronousMemcpy(void *host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 608 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
618 implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpy()
625 bool StreamExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
633 implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpy()
640 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy() function in stream_executor::StreamExecutor
665 result = implementation_->SynchronousMemcpy(host_dst, device_src, size); in SynchronousMemcpyD2H()
688 result = implementation_->SynchronousMemcpy(device_dst, host_src, size); in SynchronousMemcpyH2D()
|
| D | stream_executor_pimpl.h | 249 bool SynchronousMemcpy(DeviceMemoryBase *device_dst, const void *host_src,
256 bool SynchronousMemcpy(void *host_dst, const DeviceMemoryBase &device_src,
291 bool SynchronousMemcpy(DeviceMemoryBase *device_dst,
840 if (!parent->SynchronousMemcpy(ptr(), const_cast<const ElemT *>(&local[0]), in ScopedDeviceMemory()
|
| D | stream_executor_internal.h | 223 virtual port::Status SynchronousMemcpy(DeviceMemoryBase *gpu_dst,
225 virtual port::Status SynchronousMemcpy(void *host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 98 port::Status SynchronousMemcpy(DeviceMemoryBase *gpu_dst,
100 port::Status SynchronousMemcpy(void *host_dst,
|
| D | host_gpu_executor.cc | 163 port::Status HostExecutor::SynchronousMemcpy(DeviceMemoryBase *gpu_dst, in SynchronousMemcpy() function in stream_executor::host::HostExecutor
170 port::Status HostExecutor::SynchronousMemcpy(void *host_dst, in SynchronousMemcpy() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.h | 135 Status SynchronousMemcpy(::stream_executor::DeviceMemoryBase* device_dst,
137 Status SynchronousMemcpy(
|
| D | tpu_executor.cc | 297 Status TpuExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in tensorflow::tpu::TpuExecutor
307 Status TpuExecutor::SynchronousMemcpy( in SynchronousMemcpy() function in tensorflow::tpu::TpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 128 port::Status SynchronousMemcpy(DeviceMemoryBase* gpu_dst,
131 port::Status SynchronousMemcpy(void* host_dst,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 484 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
490 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 353 port::Status SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() function in stream_executor::__anon00950a620111::CStreamExecutor
361 port::Status SynchronousMemcpy(void* host_dst, in SynchronousMemcpy() function in stream_executor::__anon00950a620111::CStreamExecutor
|
| D | stream_executor_test.cc | 823 ASSERT_TRUE(executor->SynchronousMemcpy(&device_dst, device_src, size)); in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 576 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
582 port::Status GpuExecutor::SynchronousMemcpy(void* host_dst, in SynchronousMemcpy() function in stream_executor::gpu::GpuExecutor
|