Searched defs:device_src (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor_test.cc | 539 uint64_t size, TF_Status* const status) { in TEST_F()
551 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
593 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
600 const SP_DeviceMemoryBase* const device_src, in TEST_F()
601 uint64_t size, TF_Status* const status) { in TEST_F()
610 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
636 uint64_t size, TF_Status* const status) { in TEST_F()
646 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
|
| H A D | stream_executor_test_util.cc | 68 const SP_DeviceMemoryBase* const device_src, uint64_t size, in MemcpyDToH()
74 const SP_DeviceMemoryBase* const device_src, uint64_t size, in SyncMemcpyDToH()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.cc | 629 const DeviceMemoryBase& device_src, in SynchronousMemcpy()
644 const DeviceMemoryBase& device_src, in SynchronousMemcpy()
659 const DeviceMemoryBase& device_src, int64_t size, void* host_dst) { in SynchronousMemcpyD2H()
705 const DeviceMemoryBase& device_src, uint64_t size) { in Memcpy()
716 const DeviceMemoryBase& device_src, in MemcpyDeviceToDevice()
|
| H A D | stream_executor_pimpl.h | 285 port::Status SynchronousMemcpyD2H(const DeviceMemory<T>& device_src, in SynchronousMemcpyD2H()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executor.cc | 279 bool TpuExecutor::Memcpy(Stream* stream, void* host_dst, in Memcpy()
307 Status TpuExecutor::SynchronousMemcpy( in SynchronousMemcpy()
317 Status TpuExecutor::SynchronousMemcpyDeviceToDevice( in SynchronousMemcpyDeviceToDevice()
|
| H A D | tpu_stream.h | 56 stream_executor::DeviceMemoryBase device_src, void* host_dst, in EnqueueTransferDeviceToHost()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | while_op_test.cc | 109 uint64_t size, TF_Status* status) { in TEST_F()
|