| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executor.cc | 290 const void* host_src, uint64_t size) { in Memcpy()
298 ::stream_executor::DeviceMemoryBase* device_dst, const void* host_src, in SynchronousMemcpy()
324 bool TpuExecutor::MemcpyDeviceToDevice( in MemcpyDeviceToDevice()
|
| H A D | tpu_stream.h | 46 stream_executor::DeviceMemoryBase device_dst, const void* host_src, in EnqueueTransferHostToDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.cc | 70 const void *host_src, uint64_t size) { in Memcpy()
87 DeviceMemoryBase *dev_dst, const void *host_src, uint64_t size) { in SynchronousMemcpy()
|
| H A D | executor.h | 86 const DeviceMemoryBase &host_src, in MemcpyDeviceToDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.cc | 110 const void* host_src, uint64_t size) { in Memcpy()
164 const void* host_src, in SynchronousMemcpy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor_test_util.cc | 71 SP_DeviceMemoryBase* const device_dst, const void* host_src, in MemcpyHToD()
77 SP_DeviceMemoryBase* const device_dst, const void* host_src, in SyncMemcpyHToD()
|
| H A D | stream_executor.cc | 326 const void* host_src, uint64 size) override { in SynchronousMemcpy()
396 bool Memcpy(Stream* stream, DeviceMemoryBase* gpu_dst, const void* host_src, in Memcpy()
|
| H A D | stream_executor_test.cc | 561 TF_Status* const status) { in TEST_F()
618 const void* host_src, uint64_t size, TF_Status* const status) { in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.cc | 612 const void* host_src, uint64_t size) { in SynchronousMemcpy()
682 const void* host_src, int64_t size, DeviceMemoryBase* device_dst) { in SynchronousMemcpyH2D()
710 const void* host_src, uint64_t size) { in Memcpy()
|
| H A D | trace_listener.h | 55 const void* host_src, int64_t size, in SynchronousMemcpyH2DBegin()
|
| H A D | stream.h | 634 port::ArraySlice<ElementType> host_src, // non-absl ok in ThenMemcpyH2DQuantized()
1193 Stream &ThenMemcpyH2D(port::ArraySlice<T> host_src, // non-absl ok in ThenMemcpyH2D()
|
| H A D | stream_executor_pimpl.h | 270 port::ArraySlice<T> host_src, // non-absl ok in SynchronousMemcpyH2D()
|
| H A D | stream.cc | 908 const void *host_src, uint64_t size, dnn::QuantizedActivationMode mode, in ThenMemcpyH2DQuantized()
1999 Stream &Stream::ThenMemcpy(DeviceMemoryBase *gpu_dst, const void *host_src, in ThenMemcpy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 487 const void* host_src, in SynchronousMemcpy()
546 const void* host_src, uint64_t size) { in Memcpy()
|
| H A D | rocm_driver.cc | 858 GpuContext* context, hipDeviceptr_t gpu_dst, const void* host_src, in SynchronousMemcpyH2D()
911 const void* host_src, in AsynchronousMemcpyH2D()
|
| H A D | rocm_dnn.cc | 4516 Stream* stream, const void* host_src, int64_t size, in DoMemcpyH2DQuantized()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 619 const void* host_src, in SynchronousMemcpy()
678 const void* host_src, uint64_t size) { in Memcpy()
|
| H A D | cuda_driver.cc | 1090 const void* host_src, in SynchronousMemcpyH2D()
1171 const void* host_src, in AsynchronousMemcpyH2D()
|
| H A D | cuda_dnn.cc | 6526 Stream* stream, const void* host_src, int64_t size, in DoMemcpyH2DQuantized()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | while_op_test.cc | 117 TF_Status* const status) { in TEST_F()
|