IVGCVSW-6292 Allow profiling details to be switched off during profiling
* Add switch for network details during profiling
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I8bd49fd58f0e0255598106e9ab36806ee78391d6
diff --git a/src/armnn/Profiling.hpp b/src/armnn/Profiling.hpp
index a336a0e..372b489 100644
--- a/src/armnn/Profiling.hpp
+++ b/src/armnn/Profiling.hpp
@@ -59,6 +59,9 @@
// 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 @@
std::vector<EventPtr> m_EventSequence;
DescPtr m_ProfilingDetails = std::make_unique<ProfilingDetails>();
bool m_ProfilingEnabled;
+ bool m_EnableDetailsToStdOut;
};
// Singleton profiler manager.