aboutsummaryrefslogtreecommitdiff
path: root/DriverOptions.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2022-02-07 13:12:24 +0000
committerDavid Monahan <david.monahan@arm.com>2022-02-16 14:42:00 +0000
commitd1a947fd487eb7674271f759e43507d10e932ab0 (patch)
tree17ad41e29935a657c5025485173b4ded85c4a47c /DriverOptions.hpp
parent29520c4c785fff938586ec231cbf5e51dadbf614 (diff)
downloadandroid-nn-driver-d1a947fd487eb7674271f759e43507d10e932ab0.tar.gz
Revert "Revert "IVGCVSW-6700 Enable import aligned host memory in android-nn-driver""
This reverts commit 8069603dc44b7673b356f66517cd8b25af8080f0. * Reason for revert: Try reenable import aligned host memory in android-nn-driver * Added a check to ArmNNDriverImpl.cpp to not call ExecuteWithDummyInputs with GpuAcc * Added new android-nn-driver driver options to enable / disable Import and Export * Import is disabled by default for now due to conv2d issues * Export is enabled by default !armnn:7147 Change-Id: I91110c58ebb3931d1c458e3774944e55c1250dd8 Signed-off-by: David Monahan <David.Monahan@arm.com>
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r--DriverOptions.hpp4
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