Home
last modified time | relevance | path

Searched defs:MemcpyDeviceToDevice (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.h85 bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *pop_dst, in MemcpyDeviceToDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.cc119 bool HostExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::host::HostExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor.cc324 bool TpuExecutor::MemcpyDeviceToDevice( in MemcpyDeviceToDevice() function in tensorflow::tpu::TpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc552 bool GpuExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc410 bool MemcpyDeviceToDevice(Stream* stream, DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice() function in stream_executor::__anon702966b10111::CStreamExecutor
H A Dstream_executor_test.cc577 TEST_F(StreamExecutorTest, MemcpyDeviceToDevice) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc684 bool GpuExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc714 bool StreamExecutor::MemcpyDeviceToDevice(Stream* stream, in MemcpyDeviceToDevice() function in stream_executor::StreamExecutor