| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | light_outside_compilation_kernels_for_test.cc | 49 se::DeviceMemoryBase gpu_dst{out_tensor->data(), size}; in Compute() local
147 se::DeviceMemoryBase gpu_dst{out_tensor->data(), size_to_cpy}; in Compute() local
206 se::DeviceMemoryBase gpu_dst{out_tensor->data(), in Compute() local
292 se::DeviceMemoryBase gpu_dst{out_tensor->data(), in Compute() local
337 se::DeviceMemoryBase gpu_dst{out_tensor->data(), size_to_cpy}; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.cc | 109 bool HostExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy()
120 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
163 port::Status HostExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
178 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64_t size) { in SynchronousMemcpyDeviceToDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 486 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
501 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64_t size) { in SynchronousMemcpyDeviceToDevice()
545 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy()
553 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
| H A D | rocm_driver.cc | 858 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D()
873 GpuContext* context, hipDeviceptr_t gpu_dst, hipDeviceptr_t gpu_src, in SynchronousMemcpyD2D()
910 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyH2D()
931 hipDeviceptr_t gpu_dst, in AsynchronousMemcpyD2D()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor.cc | 325 port::Status SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
342 port::Status SynchronousMemcpyDeviceToDevice(DeviceMemoryBase* gpu_dst, in SynchronousMemcpyDeviceToDevice()
396 bool Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, const void* host_src, in Memcpy()
410 bool MemcpyDeviceToDevice(Stream* stream, DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 618 port::Status GpuExecutor::SynchronousMemcpy(DeviceMemoryBase* gpu_dst, in SynchronousMemcpy()
633 DeviceMemoryBase* gpu_dst, const DeviceMemoryBase& gpu_src, uint64_t size) { in SynchronousMemcpyDeviceToDevice()
677 bool GpuExecutor::Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, in Memcpy()
685 DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|
| H A D | cuda_driver.cc | 1089 CUdeviceptr gpu_dst, in SynchronousMemcpyH2D()
1104 CUdeviceptr gpu_dst, in SynchronousMemcpyD2D()
1170 CUdeviceptr gpu_dst, in AsynchronousMemcpyH2D()
1189 CUdeviceptr gpu_dst, in AsynchronousMemcpyD2D()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | trace_listener.h | 56 DeviceMemoryBase* gpu_dst) {} in SynchronousMemcpyH2DBegin()
|
| H A D | stream.h | 1194 DeviceMemory<T> *gpu_dst) { in ThenMemcpyH2D()
1209 Stream &ThenMemcpyD2D(DeviceMemoryBase *gpu_dst, in ThenMemcpyD2D()
|
| H A D | stream.cc | 1999 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy()
2007 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, in ThenMemcpy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executor.cc | 325 Stream* stream, ::stream_executor::DeviceMemoryBase* gpu_dst, in MemcpyDeviceToDevice()
|