aboutsummaryrefslogtreecommitdiff
path: root/delegate/include/DelegateOptions.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/include/DelegateOptions.hpp')
-rw-r--r--delegate/include/DelegateOptions.hpp35
1 files changed, 35 insertions, 0 deletions
diff --git a/delegate/include/DelegateOptions.hpp b/delegate/include/DelegateOptions.hpp
new file mode 100644
index 0000000000..0c8173d15f
--- /dev/null
+++ b/delegate/include/DelegateOptions.hpp
@@ -0,0 +1,35 @@
+//
+// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/ArmNN.hpp>
+
+#include <set>
+#include <string>
+#include <vector>
+
+namespace armnnDelegate
+{
+
+class DelegateOptions
+{
+public:
+ DelegateOptions(armnn::Compute computeDevice);
+
+ DelegateOptions(const std::vector<armnn::BackendId>& backends);
+
+ const std::vector<armnn::BackendId>& GetBackends() const { return m_Backends; }
+
+ void SetBackends(const std::vector<armnn::BackendId>& backends) { m_Backends = backends; }
+
+private:
+ /// Which backend to run Delegate on.
+ /// Examples of possible values are: CpuRef, CpuAcc, GpuAcc.
+ /// CpuRef as default.
+ std::vector<armnn::BackendId> m_Backends = { armnn::Compute::CpuRef };
+};
+
+} // namespace armnnDelegate