1 files changed, 4 insertions, 0 deletions
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