From f487486c843a38fced90229923433d09f99fc2e5 Mon Sep 17 00:00:00 2001 From: Keith Davis Date: Mon, 9 Aug 2021 16:49:18 +0100 Subject: IVGCVSW-6292 Allow profiling details to be switched off during profiling * Add switch for network details during profiling Signed-off-by: Keith Davis Change-Id: I8bd49fd58f0e0255598106e9ab36806ee78391d6 --- include/armnn/IProfiler.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'include/armnn/IProfiler.hpp') 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(); -- cgit v1.2.1