Searched defs:device_dst (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/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 | 572 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
579 SP_DeviceMemoryBase* const device_dst, in TEST_F()
581 uint64_t size, TF_Status* const status) { in TEST_F()
592 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
627 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
634 SP_DeviceMemoryBase* const device_dst, in TEST_F()
636 uint64_t size, TF_Status* const status) { in TEST_F()
645 DeviceMemoryBase device_dst(&dst_data, size); in TEST_F() local
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 611 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase* device_dst, in SynchronousMemcpy()
643 bool StreamExecutor::SynchronousMemcpy(DeviceMemoryBase* device_dst, in SynchronousMemcpy()
682 const void* host_src, int64_t size, DeviceMemoryBase* device_dst) { in SynchronousMemcpyH2D()
709 bool StreamExecutor::Memcpy(Stream* stream, DeviceMemoryBase* device_dst, in Memcpy()
715 DeviceMemoryBase* device_dst, in MemcpyDeviceToDevice()
|
| D | stream_executor_pimpl.h | 271 DeviceMemoryBase* device_dst) { in SynchronousMemcpyH2D()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | while_op_test.cc | 117 TF_Status* const status) { in TEST_F()
|