From cd639c98c347c1addae8c65eb1f4d8e75b468fdb Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Thu, 11 Feb 2021 16:57:38 +0000 Subject: IVGCVSW-5685 Add CpuAcc specific configuration option numberOfThreads * Added --number-of-threads command line option to android-nn-driver !armnn:5068 Signed-off-by: Matthew Sloyan Change-Id: I248326bf3c0355a7a17b70cb1aac7b7976820c12 --- DriverOptions.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'DriverOptions.hpp') diff --git a/DriverOptions.hpp b/DriverOptions.hpp index f3b441ca..bab611ac 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -38,6 +38,7 @@ public: int GetExitCode() const { return m_ExitCode; } const std::string& GetCachedNetworkFilePath() const { return m_CachedNetworkFilePath; } bool SaveCachedNetwork() const { return m_SaveCachedNetwork; } + unsigned int GetNumberOfThreads() const { return m_NumberOfThreads; } private: std::vector m_Backends; @@ -55,6 +56,7 @@ private: int m_ExitCode; std::string m_CachedNetworkFilePath; bool m_SaveCachedNetwork; + unsigned int m_NumberOfThreads; }; } // namespace armnn_driver -- cgit v1.2.1