Searched defs:device_src (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_stream.h | 56 stream_executor::DeviceMemoryBase device_src, void* host_dst, in EnqueueTransferDeviceToHost()
|
| 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()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 95 void* host_dst, const SP_DeviceMemoryBase* const device_src, in memcpy_dtoh()
101 const SP_DeviceMemoryBase* const device_src, in sync_memcpy_dtoh()
715 uint64_t size, TF_Status* const status) { in TEST_F()
727 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
769 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
776 const SP_DeviceMemoryBase* const device_src, in TEST_F()
777 uint64_t size, TF_Status* const status) { in TEST_F()
786 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
812 uint64_t size, TF_Status* const status) { in TEST_F()
822 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 626 const DeviceMemoryBase &device_src, in SynchronousMemcpy()
641 const DeviceMemoryBase &device_src, in SynchronousMemcpy()
656 const DeviceMemoryBase &device_src, int64 size, void *host_dst) { in SynchronousMemcpyD2H()
702 const DeviceMemoryBase &device_src, uint64 size) { in Memcpy()
713 const DeviceMemoryBase &device_src, in MemcpyDeviceToDevice()
|
| D | stream_executor_pimpl.h | 282 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H()
|