diff options
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); |