diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-25 07:46:55 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-25 07:46:55 +0000 |
| commit | f0a6dec75832604d5ab18242dc216852821a8279 (patch) | |
| tree | ff25e64c62c63975a54abd16a8bff744be70d7c0 /src/backends/reference/RefBackend.hpp | |
| parent | 16fb1a2d9c1d3d80c0f0b6ab549919fbabd2a0b9 (diff) | |
| download | armnn-f0a6dec75832604d5ab18242dc216852821a8279.tar.gz | |
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 <sadik.armagan@arm.com>
Change-Id: Iffa5b9aa9297aca4c02d923cce4636c88ac21faa
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 |