Searched defs:device_dst (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_stream.h | 46 stream_executor::DeviceMemoryBase device_dst, const void* host_src, in EnqueueTransferHostToDevice()
|
| D | tpu_executor.cc | 289 ::stream_executor::DeviceMemoryBase* device_dst, in Memcpy()
298 ::stream_executor::DeviceMemoryBase* device_dst, const void* host_src, in SynchronousMemcpy()
318 ::stream_executor::DeviceMemoryBase* device_dst, in SynchronousMemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 98 SP_DeviceMemoryBase* const device_dst, const void* host_src, in memcpy_htod()
104 SP_DeviceMemoryBase* const device_dst, in sync_memcpy_htod()
748 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
755 SP_DeviceMemoryBase* const device_dst, in TEST_F()
757 uint64_t size, TF_Status* const status) { in TEST_F()
768 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
803 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
810 SP_DeviceMemoryBase* const device_dst, in TEST_F()
812 uint64_t size, TF_Status* const status) { in TEST_F()
821 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 608 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy()
640 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase *device_dst, in SynchronousMemcpy()
679 const void *host_src, int64 size, DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D()
706 bool StreamExecutor::Memcpy(Stream *stream, DeviceMemoryBase *device_dst, in Memcpy()
712 DeviceMemoryBase *device_dst, in MemcpyDeviceToDevice()
|
| D | stream_executor_pimpl.h | 268 DeviceMemoryBase *device_dst) { in SynchronousMemcpyH2D()
|