aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon')
-rw-r--r--src/backends/neon/NeonBackend.cpp10
-rw-r--r--src/backends/neon/NeonBackend.hpp15
2 files changed, 11 insertions, 14 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp
index b496238cf3..17876753fb 100644
--- a/src/backends/neon/NeonBackend.cpp
+++ b/src/backends/neon/NeonBackend.cpp
@@ -132,16 +132,6 @@ IBackendInternal::ILayerSupportSharedPtr NeonBackend::GetLayerSupport(const Mode
return layerSupport;
}
-bool NeonBackend::HasCapability(BackendCapability capabilityClass) const
-{
- auto search = cpuAccCapabilities.find(capabilityClass);
- if (search != cpuAccCapabilities.end())
- {
- return true;
- }
- return false;
-}
-
OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph) const
{
OptimizationViews optimizationViews;
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