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 e1d25c48..ee68a945 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -42,6 +42,8 @@ public: 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; @@ -63,6 +65,8 @@ private: unsigned int m_NumberOfThreads; bool m_EnableAsyncModelExecution; unsigned int m_ArmnnNumberOfThreads; + bool m_EnableImport; + bool m_EnableExport; }; } // namespace armnn_driver |