Home
last modified time | relevance | path

Searched refs:SP_DeviceMemoryBase (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h141 typedef struct SP_DeviceMemoryBase { struct
149 } SP_DeviceMemoryBase; argument
152 TF_OFFSET_OF_END(SP_DeviceMemoryBase, payload)
246 SP_DeviceMemoryBase* mem);
250 void (*deallocate)(const SP_Device* device, SP_DeviceMemoryBase* memory);
342 const SP_DeviceMemoryBase* device_src, uint64_t size,
348 SP_DeviceMemoryBase* device_dst, const void* host_src,
354 SP_DeviceMemoryBase* device_dst,
355 const SP_DeviceMemoryBase* device_src, uint64_t size,
361 const SP_DeviceMemoryBase* device_src, uint64_t size,
[all …]
Dstream_executor.cc126 port::Status ValidateSPDeviceMemoryBase(const SP_DeviceMemoryBase& mem) { in ValidateSPDeviceMemoryBase()
127 VALIDATE_STRUCT_SIZE(SP_DeviceMemoryBase, mem, in ValidateSPDeviceMemoryBase()
206 SP_DeviceMemoryBase DeviceMemoryBaseToC(const DeviceMemoryBase* mem) { in DeviceMemoryBaseToC()
207 SP_DeviceMemoryBase device_memory_base{SP_DEVICE_MEMORY_BASE_STRUCT_SIZE}; in DeviceMemoryBaseToC()
216 DeviceMemoryBase DeviceMemoryBaseFromC(const SP_DeviceMemoryBase& mem) { in DeviceMemoryBaseFromC()
262 SP_DeviceMemoryBase mem = {SP_DEVICE_MEMORY_BASE_STRUCT_SIZE}; in Allocate()
279 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(mem); in Deallocate()
356 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(gpu_dst); in SynchronousMemcpy()
365 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(&gpu_src); in SynchronousMemcpy()
374 SP_DeviceMemoryBase device_mem_dst = DeviceMemoryBaseToC(gpu_dst); in SynchronousMemcpyDeviceToDevice()
[all …]
Dstream_executor_test.cc50 int64_t memory_space, SP_DeviceMemoryBase* const mem) {} in allocate()
51 void deallocate(const SP_Device* const device, SP_DeviceMemoryBase* const mem) { in deallocate()
95 void* host_dst, const SP_DeviceMemoryBase* const device_src, in memcpy_dtoh()
98 SP_DeviceMemoryBase* const device_dst, const void* host_src, in memcpy_htod()
101 const SP_DeviceMemoryBase* const device_src, in sync_memcpy_dtoh()
104 SP_DeviceMemoryBase* const device_dst, in sync_memcpy_htod()
346 int64_t memory_space, SP_DeviceMemoryBase* const mem) { in TEST_F()
352 SP_DeviceMemoryBase* const mem) { in TEST_F()
714 const SP_DeviceMemoryBase* const device_src, in TEST_F()
735 SP_DeviceMemoryBase* const device_dst, in TEST_F()
[all …]