diff options
Diffstat (limited to 'delegate/include/DelegateOptions.hpp')
-rw-r--r-- | delegate/include/DelegateOptions.hpp | 35 |
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 |