diff options
Diffstat (limited to 'include/armnn/IProfiler.hpp')
-rw-r--r-- | include/armnn/IProfiler.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/include/armnn/IProfiler.hpp b/include/armnn/IProfiler.hpp index ac422b714f..1d350855f6 100644 --- a/include/armnn/IProfiler.hpp +++ b/include/armnn/IProfiler.hpp @@ -16,6 +16,8 @@ class ProfilerImpl; class BackendId; class Instrument; class Event; +struct WorkloadInfo; + class IProfiler { public: @@ -41,12 +43,22 @@ public: IProfiler(); private: + using InstrumentPtr = std::unique_ptr<Instrument>; + + template<typename DescriptorType> + void AddLayerDetails(const std::string& name, + const DescriptorType& desc, + const WorkloadInfo& infos); + Event* BeginEvent(const BackendId& backendId, const std::string& label, std::vector<InstrumentPtr>&& instruments); + std::unique_ptr<ProfilerImpl> pProfilerImpl; + friend class ScopedProfilingEvent; + friend class ScopedProfilingUpdateDescriptions; // Friend functions for unit testing, see ProfilerTests.cpp. friend size_t GetProfilerEventSequenceSize(armnn::IProfiler* profiler); |