diff options
Diffstat (limited to 'src/backends/reference/RefBackend.hpp')
-rw-r--r-- | src/backends/reference/RefBackend.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp index 92d392dde6..c92936ca0c 100644 --- a/src/backends/reference/RefBackend.hpp +++ b/src/backends/reference/RefBackend.hpp @@ -9,6 +9,10 @@ namespace armnn { +const std::set<armnn::BackendCapability> backendCapabilities { + armnn::BackendCapability::NonConstWeights, +}; + class RefBackend : public IBackendInternal { public: @@ -39,6 +43,8 @@ public: std::vector<ITensorHandleFactory::FactoryId> GetHandleFactoryPreferences() const override; void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) override; + + bool HasCapability(BackendCapability capabilityClass) const override; }; } // namespace armnn |