aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/IProfiler.hpp
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2021-08-09 16:49:18 +0100
committerMatthew Sloyan <matthew.sloyan@arm.com>2021-08-10 14:07:27 +0000
commitf487486c843a38fced90229923433d09f99fc2e5 (patch)
tree63a2268196279a5c97c3b9dea86024f0bf4a751e /include/armnn/IProfiler.hpp
parentbcd860a30eba22bb2ba0943ad705734ce0ec5b23 (diff)
downloadarmnn-f487486c843a38fced90229923433d09f99fc2e5.tar.gz
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
Diffstat (limited to 'include/armnn/IProfiler.hpp')
-rw-r--r--include/armnn/IProfiler.hpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/include/armnn/IProfiler.hpp b/include/armnn/IProfiler.hpp
index 5ce42017d9..5fbb67000a 100644
--- a/include/armnn/IProfiler.hpp
+++ b/include/armnn/IProfiler.hpp
@@ -39,6 +39,10 @@ public:
/// @param [out] outStream The stream where to write the profiling results to.
void Print(std::ostream& outStream) const;
+ /// Print out details of each layer within the network that possesses a descriptor.
+ /// Also outputs tensor info.
+ void EnableNetworkDetailsToStdOut();
+
~IProfiler();
IProfiler();