diff options
| author | Keith Davis <keith.davis@arm.com> | 2021-08-18 17:14:05 +0100 |
|---|---|---|
| committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2021-08-20 09:10:22 +0000 |
| commit | 4914d0c1f4cbb01fe276a7093af4cff13270b74a (patch) | |
| tree | ce6c4364bb596803148833af535215aa6ab6de90 /src/armnn/Profiling.hpp | |
| parent | e369dbddae66c58f4b444c4b25871c10af19ed9d (diff) | |
| download | armnn-4914d0c1f4cbb01fe276a7093af4cff13270b74a.tar.gz | |
IVGCVSW-6249 Add ProfilingDetails Macros to all workloads in Ref, Neon, CL
* Add functionality to only output network details in ExNet
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I0c45e67193f308ce7b86f1bb1a918a266fefba2e
Diffstat (limited to 'src/armnn/Profiling.hpp')
| -rw-r--r-- | src/armnn/Profiling.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/Profiling.hpp b/src/armnn/Profiling.hpp index 372b489abf..42d7f4d638 100644 --- a/src/armnn/Profiling.hpp +++ b/src/armnn/Profiling.hpp @@ -60,7 +60,7 @@ public: bool IsProfilingEnabled(); // Enables outputting the layer descriptors and infos to stdout - void EnableNetworkDetailsToStdOut(); + void EnableNetworkDetailsToStdOut(ProfilingDetailsMethod detailsMethod); // Increments the event tag, allowing grouping of events in a user-defined manner (e.g. per inference). void UpdateEventTag(); @@ -102,7 +102,8 @@ public: std::vector<EventPtr> m_EventSequence; DescPtr m_ProfilingDetails = std::make_unique<ProfilingDetails>(); bool m_ProfilingEnabled; - bool m_EnableDetailsToStdOut; + ProfilingDetailsMethod m_DetailsToStdOutMethod; + }; // Singleton profiler manager. |