Home
last modified time | relevance | path

Searched refs:MemcpyDeviceToDevice (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h79 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *gpu_dst,
Dhost_gpu_executor.cc119 bool HostExecutor::MemcpyDeviceToDevice(Stream *stream, in MemcpyDeviceToDevice() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h86 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *pop_dst, in MemcpyDeviceToDevice() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h127 bool MemcpyDeviceToDevice(Stream* stream,
Dtpu_executor.cc324 bool TpuExecutor::MemcpyDeviceToDevice( in MemcpyDeviceToDevice() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h243 virtual bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *gpu_dst,
Dstream_executor_pimpl.cc711 bool StreamExecutor::MemcpyDeviceToDevice(Stream *stream, in MemcpyDeviceToDevice() function in stream_executor::StreamExecutor
715 return implementation_->MemcpyDeviceToDevice(stream, device_dst, device_src, in MemcpyDeviceToDevice()
Dstream_executor_pimpl.h584 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *device_dst,
Dstream.cc4519 CheckError(parent_->MemcpyDeviceToDevice(this, gpu_dst, gpu_src, size)); in ThenMemcpy()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h152 bool MemcpyDeviceToDevice(Stream* stream, DeviceMemoryBase* gpu_dst,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc549 bool GpuExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc423 bool MemcpyDeviceToDevice(Stream* stream, DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice() function in stream_executor::__anon00950a620111::CStreamExecutor
Dstream_executor_test.cc753 TEST_F(StreamExecutorTest, MemcpyDeviceToDevice) { in TEST_F() argument
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc641 bool GpuExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::gpu::GpuExecutor