diff options
Diffstat (limited to 'src/backends/neon/NeonBackendModelContext.hpp')
| -rw-r--r-- | src/backends/neon/NeonBackendModelContext.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/neon/NeonBackendModelContext.hpp b/src/backends/neon/NeonBackendModelContext.hpp index 430ae45a6f..e736efc1d8 100644 --- a/src/backends/neon/NeonBackendModelContext.hpp +++ b/src/backends/neon/NeonBackendModelContext.hpp @@ -14,6 +14,8 @@ namespace armnn /// - "FastMathEnabled"\n /// Using the fast_math flag can lead to performance improvements in fp32 and fp16 layers but may result in\n /// results with reduced or different precision. The fast_math flag will not have any effect on int8 performance. +/// - "NumberOfThreads"\n +/// Specify the number of threads used by the CpuAcc backend. class NeonBackendModelContext : public IBackendModelContext { public: @@ -21,8 +23,11 @@ public: bool IsFastMathEnabled() const; + unsigned int GetNumberOfThreads() const; + private: bool m_IsFastMathEnabled; + unsigned int m_NumberOfThreads; }; } // namespace armnn
\ No newline at end of file |