diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-05-26 18:38:12 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-06-09 17:18:10 +0100 |
commit | b9af86ea42568ade799ee5529137e4756977b6c6 (patch) | |
tree | 261003078fd2191b22ee7465e07668cbed666553 /src/backends/neon/NeonBackend.hpp | |
parent | 5b1bcc93820b442bc4035c1e030a8d4a0983df91 (diff) | |
download | armnn-b9af86ea42568ade799ee5529137e4756977b6c6.tar.gz |
IVGCVSW-5855 Refactor the reporting of capabilities from backends
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I05fc331a8e91bdcb6b8a2f32cfb555060fc5d797
Diffstat (limited to 'src/backends/neon/NeonBackend.hpp')
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 34606f992c..eae57320cc 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -9,9 +9,13 @@ namespace armnn { -const std::set<armnn::BackendCapability> cpuAccCapabilities { - // add new capabilities here.. -}; +// add new capabilities here.. +const BackendCapabilities cpuAccCapabilities("GpuAcc", + { + {"NonConstWeights", false}, + {"AsyncExecution", false} + }); + class NeonBackend : public IBackendInternal { @@ -52,7 +56,10 @@ public: IBackendInternal::IBackendSpecificModelContextPtr CreateBackendSpecificModelContext( const ModelOptions& modelOptions) const override; - bool HasCapability(BackendCapability capabilityClass) const override; + BackendCapabilities GetCapabilities() const override + { + return cpuAccCapabilities; + }; }; } // namespace armnn |