diff options
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.cpp')
-rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index 0add6f594d..3638212c75 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -4,6 +4,7 @@ // #include "NeonLayerSupport.hpp" +#include "NeonBackendId.hpp" #include <armnn/Descriptors.hpp> #include <armnn/InternalTypes.hpp> @@ -11,6 +12,8 @@ #include <armnn/Tensor.hpp> #include <armnn/Types.hpp> +#include <backends/LayerSupportRegistry.hpp> + #include <boost/core/ignore_unused.hpp> #ifdef ARMCOMPUTENEON_ENABLED @@ -37,6 +40,21 @@ namespace armnn namespace { +ILayerSupportSharedPtr GetLayerSupportPointer() +{ + static ILayerSupportSharedPtr instance{new NeonLayerSupport}; + return instance; +} + +static StaticRegistryInitializer<LayerSupportRegistry> g_RegisterHelper{ + LayerSupportRegistryInstance(), + NeonBackendId(), + []() + { + return GetLayerSupportPointer(); + } +}; + bool IsNeonBackendSupported(Optional<std::string&> reasonIfUnsupported) { #if ARMCOMPUTENEON_ENABLED |