diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:31:35 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:31:35 +0100 |
commit | ce3e84a8d449cbf31cee57e30f0eef6a96c0ce94 (patch) | |
tree | 77a769e27879fb712027c990071b061c5e0e3b60 /DriverOptions.hpp | |
parent | deb3bdbe028a59da0759dd7a560387d03a11d322 (diff) | |
download | android-nn-driver-ce3e84a8d449cbf31cee57e30f0eef6a96c0ce94.tar.gz |
Release 18.08
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r-- | DriverOptions.hpp | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp new file mode 100644 index 00000000..4a378b26 --- /dev/null +++ b/DriverOptions.hpp @@ -0,0 +1,43 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// + +#pragma once + +#include <armnn/ArmNN.hpp> + +#include <set> +#include <string> + +namespace armnn_driver +{ + +class DriverOptions +{ +public: + DriverOptions(armnn::Compute computeDevice); + DriverOptions(int argc, char** argv); + DriverOptions(DriverOptions&& other) = default; + + armnn::Compute GetComputeDevice() const { return m_ComputeDevice; } + bool IsVerboseLoggingEnabled() const { return m_VerboseLogging; } + const std::string& GetRequestInputsAndOutputsDumpDir() const { return m_RequestInputsAndOutputsDumpDir; } + const std::set<unsigned int>& GetForcedUnsupportedOperations() const { return m_ForcedUnsupportedOperations; } + const std::string& GetClTunedParametersFile() const { return m_ClTunedParametersFile; } + armnn::IGpuAccTunedParameters::Mode GetClTunedParametersMode() const { return m_ClTunedParametersMode; } + bool IsGpuProfilingEnabled() const { return m_EnableGpuProfiling; } + bool GetFp16Enabled() const { return m_fp16Enabled; } + +private: + armnn::Compute m_ComputeDevice; + bool m_VerboseLogging; + std::string m_RequestInputsAndOutputsDumpDir; + std::set<unsigned int> m_ForcedUnsupportedOperations; + std::string m_ClTunedParametersFile; + armnn::IGpuAccTunedParameters::Mode m_ClTunedParametersMode; + bool m_EnableGpuProfiling; + bool m_fp16Enabled; +}; + +} // namespace armnn_driver |