diff options
Diffstat (limited to 'src/backends/neon/NeonBackendModelContext.cpp')
-rw-r--r-- | src/backends/neon/NeonBackendModelContext.cpp | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackendModelContext.cpp b/src/backends/neon/NeonBackendModelContext.cpp index 2be71e5ded..270592e94d 100644 --- a/src/backends/neon/NeonBackendModelContext.cpp +++ b/src/backends/neon/NeonBackendModelContext.cpp @@ -17,13 +17,22 @@ bool ParseBool(const armnn::BackendOptions::Var& value, bool defaultValue) return defaultValue; } +unsigned int ParseUnsignedInt(const armnn::BackendOptions::Var& value, unsigned int defaultValue) +{ + if (value.IsUnsignedInt()) + { + return value.AsUnsignedInt(); + } + return defaultValue; +} + } // namespace anonymous namespace armnn { NeonBackendModelContext::NeonBackendModelContext(const ModelOptions& modelOptions) - : m_IsFastMathEnabled(false) + : m_IsFastMathEnabled(false), m_NumberOfThreads(0) { if (!modelOptions.empty()) { @@ -33,6 +42,10 @@ NeonBackendModelContext::NeonBackendModelContext(const ModelOptions& modelOption { m_IsFastMathEnabled |= ParseBool(value, false); } + if (name == "NumberOfThreads") + { + m_NumberOfThreads |= ParseUnsignedInt(value, 0); + } }); } } @@ -42,4 +55,9 @@ bool NeonBackendModelContext::IsFastMathEnabled() const return m_IsFastMathEnabled; } +unsigned int NeonBackendModelContext::GetNumberOfThreads() const +{ + return m_NumberOfThreads; +} + } // namespace armnn
\ No newline at end of file |