From f0a6dec75832604d5ab18242dc216852821a8279 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 25 Mar 2021 07:46:55 +0000 Subject: IVGCVSW-5736 and IVGCVSW-5743 'NonConstWeights: Update front-end and TfLiteDelegate support for FullyConnected Operator' * Added front-end support for non-const weights for FULLY_CONNECTED operator * Added FULLY_CONNECTED end-to-end test * Updated FULLY_CONNECTED operator support in TfLite Arm NN Delegate for non-const weights * Updated the version numbers Signed-off-by: Sadik Armagan Change-Id: Iffa5b9aa9297aca4c02d923cce4636c88ac21faa --- src/backends/reference/RefBackend.hpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/backends/reference/RefBackend.hpp') 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 backendCapabilities { + armnn::BackendCapability::NonConstWeights, +}; + class RefBackend : public IBackendInternal { public: @@ -39,6 +43,8 @@ public: std::vector GetHandleFactoryPreferences() const override; void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) override; + + bool HasCapability(BackendCapability capabilityClass) const override; }; } // namespace armnn -- cgit v1.2.1