diff options
Diffstat (limited to 'DriverOptions.hpp')
| -rw-r--r-- | DriverOptions.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp index 8b3f5743..ee68a945 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -40,6 +40,10 @@ 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; }; + bool isImportEnabled() const { return m_EnableImport; }; + bool isExportEnabled() const { return m_EnableExport; }; private: std::vector<armnn::BackendId> m_Backends; @@ -59,6 +63,10 @@ private: std::string m_CachedNetworkFilePath; bool m_SaveCachedNetwork; unsigned int m_NumberOfThreads; + bool m_EnableAsyncModelExecution; + unsigned int m_ArmnnNumberOfThreads; + bool m_EnableImport; + bool m_EnableExport; }; } // namespace armnn_driver |