aboutsummaryrefslogtreecommitdiff
path: root/DriverOptions.hpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2021-02-11 16:57:38 +0000
committerMatthew Sloyan <matthew.sloyan@arm.com>2021-02-12 18:15:12 +0000
commitcd639c98c347c1addae8c65eb1f4d8e75b468fdb (patch)
treed6a4a1ea0d649df94a30272c7247f43071f4e8b1 /DriverOptions.hpp
parent21469d2498017241d4caac4097479e249d78bf5d (diff)
downloadandroid-nn-driver-cd639c98c347c1addae8c65eb1f4d8e75b468fdb.tar.gz
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 <matthew.sloyan@arm.com> Change-Id: I248326bf3c0355a7a17b70cb1aac7b7976820c12
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r--DriverOptions.hpp2
1 files changed, 2 insertions, 0 deletions
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<armnn::BackendId> m_Backends;
@@ -55,6 +56,7 @@ private:
int m_ExitCode;
std::string m_CachedNetworkFilePath;
bool m_SaveCachedNetwork;
+ unsigned int m_NumberOfThreads;
};
} // namespace armnn_driver