diff options
Diffstat (limited to 'src/armnn/WallClockTimer.hpp')
-rw-r--r-- | src/armnn/WallClockTimer.hpp | 4 |
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 |