aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonBackendModelContext.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/NeonBackendModelContext.cpp')
-rw-r--r--src/backends/neon/NeonBackendModelContext.cpp20
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