diff options
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r-- | DriverOptions.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp index 8b3f5743..e1d25c48 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -40,6 +40,8 @@ public: const std::string& GetCachedNetworkFilePath() const { return m_CachedNetworkFilePath; } bool SaveCachedNetwork() const { return m_SaveCachedNetwork; } unsigned int GetNumberOfThreads() const { return m_NumberOfThreads; } + bool isAsyncModelExecutionEnabled() const { return m_EnableAsyncModelExecution; }; + unsigned int getNoOfArmnnThreads() const { return m_ArmnnNumberOfThreads; }; private: std::vector<armnn::BackendId> m_Backends; @@ -59,6 +61,8 @@ private: std::string m_CachedNetworkFilePath; bool m_SaveCachedNetwork; unsigned int m_NumberOfThreads; + bool m_EnableAsyncModelExecution; + unsigned int m_ArmnnNumberOfThreads; }; } // namespace armnn_driver |