aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/BackendHelper.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-03-25 07:46:55 +0000
committerSadik Armagan <sadik.armagan@arm.com>2021-03-25 07:46:55 +0000
commitf0a6dec75832604d5ab18242dc216852821a8279 (patch)
treeff25e64c62c63975a54abd16a8bff744be70d7c0 /include/armnn/BackendHelper.hpp
parent16fb1a2d9c1d3d80c0f0b6ab549919fbabd2a0b9 (diff)
downloadarmnn-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 'include/armnn/BackendHelper.hpp')
-rw-r--r--include/armnn/BackendHelper.hpp10
1 files changed, 9 insertions, 1 deletions
diff --git a/include/armnn/BackendHelper.hpp b/include/armnn/BackendHelper.hpp
index a562f60c23..41bb5f9c3a 100644
--- a/include/armnn/BackendHelper.hpp
+++ b/include/armnn/BackendHelper.hpp
@@ -7,6 +7,7 @@
#include <armnn/BackendId.hpp>
#include <armnn/ILayerSupport.hpp>
+#include <armnn/Types.hpp>
namespace armnn
{
@@ -19,7 +20,10 @@ class LayerSupportHandle
{
public:
explicit LayerSupportHandle(std::shared_ptr<ILayerSupport> layerSupport)
- : m_LayerSupport(std::move(layerSupport)) {};
+ : m_LayerSupport(std::move(layerSupport)), m_BackendId(Compute::Undefined) {};
+
+ explicit LayerSupportHandle(std::shared_ptr<ILayerSupport> layerSupport, const BackendId& backendId)
+ : m_LayerSupport(std::move(layerSupport)), m_BackendId(backendId) {};
bool IsBackendRegistered() const;
@@ -422,9 +426,13 @@ public:
private:
std::shared_ptr<ILayerSupport> m_LayerSupport;
+ const BackendId m_BackendId;
};
/// Convenience function to retrieve the ILayerSupportHandle for a backend
LayerSupportHandle GetILayerSupportByBackendId(const armnn::BackendId& backend);
+/// Convenience function to check a capability on a backend
+bool IsCapabilitySupported(const armnn::BackendId& backend, armnn::BackendCapability capability);
+
}