diff options
Diffstat (limited to 'src/armnn/Profiling.hpp')
-rw-r--r-- | src/armnn/Profiling.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/Profiling.hpp b/src/armnn/Profiling.hpp index a336a0ee2a..372b489abf 100644 --- a/src/armnn/Profiling.hpp +++ b/src/armnn/Profiling.hpp @@ -59,6 +59,9 @@ public: // Checks if profiling is enabled. bool IsProfilingEnabled(); + // Enables outputting the layer descriptors and infos to stdout + void EnableNetworkDetailsToStdOut(); + // Increments the event tag, allowing grouping of events in a user-defined manner (e.g. per inference). void UpdateEventTag(); @@ -99,6 +102,7 @@ public: std::vector<EventPtr> m_EventSequence; DescPtr m_ProfilingDetails = std::make_unique<ProfilingDetails>(); bool m_ProfilingEnabled; + bool m_EnableDetailsToStdOut; }; // Singleton profiler manager. |