From 69851b525b9040ee7bf4b796efe74d473bc4e321 Mon Sep 17 00:00:00 2001
From: Nina Drozd <nina.drozd@arm.com>
Date: Fri, 21 Sep 2018 18:42:09 +0100
Subject: IVGCVSW-1821 - update NEON workload utils to use timers in correct
 order, updated units used in NeonTimer

Change-Id: I593af42bd2930dd9d147354b706087e3ac260fe9
---
 src/armnn/WallClockTimer.hpp | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/armnn/WallClockTimer.hpp')

diff --git a/src/armnn/WallClockTimer.hpp b/src/armnn/WallClockTimer.hpp
index 5e88382015..09cc514eae 100644
--- a/src/armnn/WallClockTimer.hpp
+++ b/src/armnn/WallClockTimer.hpp
@@ -42,6 +42,8 @@ public:
     // Get the name of the timer
     const char* GetName() const override;
 
+    void SetScaleFactor(Measurement::Unit measurementUnit);
+
     // Get the recorded measurements
     std::vector<Measurement> GetMeasurements() const override;
 
@@ -58,6 +60,8 @@ public:
 private:
     clock::time_point m_Start;
     clock::time_point m_Stop;
+    float m_ScaleFactor = 1.f;
+    Measurement::Unit m_Unit = Measurement::Unit::TIME_MS;
 };
 
 } //namespace armnn
-- 
cgit v1.2.1