diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-04 13:33:10 +0100 |
commit | d5fd9767d426ca465eb660be062eecce51ad1097 (patch) | |
tree | b2a2b684c9072dda7b52b40e579a5b0c4aae3c21 /DriverOptions.hpp | |
parent | 16196e267833178dae62926ff090c50ec6813ad4 (diff) | |
download | android-nn-driver-d5fd9767d426ca465eb660be062eecce51ad1097.tar.gz |
IVGCVSW-2886 Support multiple backends in Android driver
Change-Id: I4abe1f750801911570b6dc65c187b828c5929b5f
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r-- | DriverOptions.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp index 7271ac16..637ccd64 100644 --- a/DriverOptions.hpp +++ b/DriverOptions.hpp @@ -9,6 +9,7 @@ #include <set> #include <string> +#include <vector> namespace armnn_driver { @@ -17,10 +18,11 @@ class DriverOptions { public: DriverOptions(armnn::Compute computeDevice, bool fp16Enabled = false); + DriverOptions(const std::vector<armnn::BackendId>& backends, bool fp16Enabled); DriverOptions(int argc, char** argv); DriverOptions(DriverOptions&& other) = default; - armnn::Compute GetComputeDevice() const { return m_ComputeDevice; } + const std::vector<armnn::BackendId>& GetBackends() const { return m_Backends; } bool IsVerboseLoggingEnabled() const { return m_VerboseLogging; } const std::string& GetRequestInputsAndOutputsDumpDir() const { return m_RequestInputsAndOutputsDumpDir; } const std::set<unsigned int>& GetForcedUnsupportedOperations() const { return m_ForcedUnsupportedOperations; } @@ -30,7 +32,7 @@ public: bool GetFp16Enabled() const { return m_fp16Enabled; } private: - armnn::Compute m_ComputeDevice; + std::vector<armnn::BackendId> m_Backends; bool m_VerboseLogging; std::string m_RequestInputsAndOutputsDumpDir; std::set<unsigned int> m_ForcedUnsupportedOperations; |