aboutsummaryrefslogtreecommitdiff
path: root/DriverOptions.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-04-04 13:33:10 +0100
committerNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-04-04 13:33:10 +0100
commitd5fd9767d426ca465eb660be062eecce51ad1097 (patch)
treeb2a2b684c9072dda7b52b40e579a5b0c4aae3c21 /DriverOptions.hpp
parent16196e267833178dae62926ff090c50ec6813ad4 (diff)
downloadandroid-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.hpp6
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;