Home
last modified time | relevance | path

Searched refs:SP_Device (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h154 typedef struct SP_Device { struct
183 } SP_Device; argument
185 #define SP_DEVICE_STRUCT_SIZE TF_OFFSET_OF_END(SP_Device, pci_bus_id)
192 SP_Device* device; // Input/output, struct_size set by TF for plugin to read.
208 int32_t (*get_numa_node)(const SP_Device* device);
214 int64_t (*get_memory_bandwidth)(const SP_Device* device);
220 double (*get_gflops)(const SP_Device* device);
245 void (*allocate)(const SP_Device* device, uint64_t size, int64_t memory_space,
250 void (*deallocate)(const SP_Device* device, SP_DeviceMemoryBase* memory);
255 void* (*host_memory_allocate)(const SP_Device* device, uint64_t size);
[all …]
Dstream_executor_test.cc49 void allocate(const SP_Device* const device, uint64_t size, in allocate()
51 void deallocate(const SP_Device* const device, SP_DeviceMemoryBase* const mem) { in deallocate()
53 void* host_memory_allocate(const SP_Device* const device, uint64_t size) { in host_memory_allocate()
56 void host_memory_deallocate(const SP_Device* const device, void* mem) {} in host_memory_deallocate()
57 TF_Bool get_allocator_stats(const SP_Device* const device, in get_allocator_stats()
61 TF_Bool device_memory_usage(const SP_Device* const device, int64_t* const free, in device_memory_usage()
65 void create_stream(const SP_Device* const device, SP_Stream* stream, in create_stream()
69 void destroy_stream(const SP_Device* const device, SP_Stream stream) {} in destroy_stream()
70 void create_stream_dependency(const SP_Device* const device, in create_stream_dependency()
73 void get_stream_status(const SP_Device* const device, SP_Stream stream, in get_stream_status()
[all …]
Dstream_executor_internal.h96 CStream(SP_Device* device, SP_StreamExecutor* stream_executor) in CStream()
119 SP_Device* device_;
126 CEvent(SP_Device* device, SP_StreamExecutor* stream_executor) in CEvent()
155 SP_Device* device_;
162 CTimer(SP_Device* device, SP_StreamExecutor* stream_executor, in CTimer()
194 SP_Device* device_;
Dstream_executor.cc133 port::Status ValidateSPDevice(const SP_Device& device) { in ValidateSPDevice()
134 VALIDATE_STRUCT_SIZE(SP_Device, device, SP_DEVICE_STRUCT_SIZE); in ValidateSPDevice()
239 explicit CStreamExecutor(SP_Device device, SP_DeviceFns* device_fns, in CStreamExecutor()
662 SP_Device device_;
727 SP_Device device{SP_DEVICE_STRUCT_SIZE}; in GetUncachedExecutor()