Searched defs:device_src (Results 1 – 6 of 6) 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 | 537 uint64_t size, TF_Status* const status) { in TEST_F()
549 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
591 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
598 const SP_DeviceMemoryBase* const device_src, in TEST_F()
599 uint64_t size, TF_Status* const status) { in TEST_F()
608 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
634 uint64_t size, TF_Status* const status) { in TEST_F()
644 DeviceMemoryBase device_src(&src_data, size); in TEST_F() local
|
| 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()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 633 const DeviceMemoryBase &device_src, in SynchronousMemcpy()
648 const DeviceMemoryBase &device_src, in SynchronousMemcpy()
663 const DeviceMemoryBase &device_src, int64_t size, void *host_dst) { in SynchronousMemcpyD2H()
709 const DeviceMemoryBase &device_src, uint64 size) { in Memcpy()
720 const DeviceMemoryBase &device_src, in MemcpyDeviceToDevice()
|
| D | stream_executor_pimpl.h | 285 port::Status SynchronousMemcpyD2H(const DeviceMemory<T> &device_src, in SynchronousMemcpyD2H()
|