Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc145 port::Status ValidateSPStreamExecutor(const SP_StreamExecutor& se, in ValidateSPStreamExecutor()
147 VALIDATE_STRUCT_SIZE(SP_StreamExecutor, se, SP_STREAM_EXECUTOR_STRUCT_SIZE); in ValidateSPStreamExecutor()
148 VALIDATE_MEMBER(SP_StreamExecutor, se, allocate); in ValidateSPStreamExecutor()
149 VALIDATE_MEMBER(SP_StreamExecutor, se, deallocate); in ValidateSPStreamExecutor()
150 VALIDATE_MEMBER(SP_StreamExecutor, se, get_allocator_stats); in ValidateSPStreamExecutor()
151 VALIDATE_MEMBER(SP_StreamExecutor, se, host_memory_allocate); in ValidateSPStreamExecutor()
152 VALIDATE_MEMBER(SP_StreamExecutor, se, host_memory_deallocate); in ValidateSPStreamExecutor()
154 VALIDATE_MEMBER(SP_StreamExecutor, se, unified_memory_allocate); in ValidateSPStreamExecutor()
155 VALIDATE_MEMBER(SP_StreamExecutor, se, unified_memory_deallocate); in ValidateSPStreamExecutor()
157 VALIDATE_MEMBER(SP_StreamExecutor, se, device_memory_usage); in ValidateSPStreamExecutor()
[all …]
Dstream_executor_internal.h54 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor,
87 SP_StreamExecutor stream_executor_;
96 CStream(SP_Device* device, SP_StreamExecutor* stream_executor) in CStream()
120 SP_StreamExecutor* stream_executor_;
126 CEvent(SP_Device* device, SP_StreamExecutor* stream_executor) in CEvent()
156 SP_StreamExecutor* stream_executor_;
162 CTimer(SP_Device* device, SP_StreamExecutor* stream_executor, in CTimer()
195 SP_StreamExecutor* stream_executor_;
Dstream_executor.h235 typedef struct SP_StreamExecutor { struct
400 } SP_StreamExecutor; argument
403 TF_OFFSET_OF_END(SP_StreamExecutor, host_callback)
409 SP_StreamExecutor* stream_executor; // output, to be filled by plugin
472 SP_StreamExecutor* stream_executor);
Dstream_executor_test.cc117 void PopulateDefaultStreamExecutor(SP_StreamExecutor* se) { in PopulateDefaultStreamExecutor()
173 SP_StreamExecutor* se) {} in destroy_stream_executor()
339 SP_StreamExecutor se_;