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 /ModelToINetworkConverter.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 'ModelToINetworkConverter.hpp')
-rw-r--r-- | ModelToINetworkConverter.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ModelToINetworkConverter.hpp b/ModelToINetworkConverter.hpp index a3758fd5..e78c5f02 100644 --- a/ModelToINetworkConverter.hpp +++ b/ModelToINetworkConverter.hpp @@ -11,6 +11,7 @@ #include <armnn/ArmNN.hpp> #include <set> +#include <vector> namespace armnn_driver { @@ -30,7 +31,7 @@ class ModelToINetworkConverter public: using HalModel = typename HalPolicy::Model; - ModelToINetworkConverter(armnn::Compute compute, + ModelToINetworkConverter(const std::vector<armnn::BackendId>& backends, const HalModel& model, const std::set<unsigned int>& forcedUnsupportedOperations); |