diff options
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 909df75445..253d9c2294 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -4,11 +4,14 @@ // #include "RefLayerSupport.hpp" +#include "RefBackendId.hpp" #include <armnn/InternalTypes.hpp> #include <armnn/LayerSupportCommon.hpp> #include <armnn/Types.hpp> +#include <backends/LayerSupportRegistry.hpp> + #include <boost/core/ignore_unused.hpp> using namespace boost; @@ -19,6 +22,21 @@ namespace armnn namespace { +ILayerSupportSharedPtr GetLayerSupportPointer() +{ + static ILayerSupportSharedPtr instance{new RefLayerSupport}; + return instance; +} + +static StaticRegistryInitializer<LayerSupportRegistry> g_RegisterHelper{ + LayerSupportRegistryInstance(), + RefBackendId(), + []() + { + return GetLayerSupportPointer(); + } +}; + template<typename Float32Func, typename Uint8Func, typename ... Params> bool IsSupportedForDataTypeRef(Optional<std::string&> reasonIfUnsupported, DataType dataType, |