diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 23 | ||||
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 8 |
2 files changed, 24 insertions, 7 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index 1137145aa2..b4f1897704 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -4,11 +4,26 @@ // #include "NeonBackend.hpp" +#include <backends/BackendRegistry.hpp> +#include <boost/cast.hpp> namespace armnn { -const std::string NeonBackend::s_Id = "arm_compute_neon"; +namespace +{ + +static const std::string s_Id = "CpuAcc"; + +static BackendRegistry::Helper g_RegisterHelper{ + s_Id, + []() + { + return IBackendUniquePtr(new NeonBackend, &NeonBackend::Destroy); + } +}; + +} const std::string& NeonBackend::GetId() const { @@ -22,8 +37,12 @@ const ILayerSupport& NeonBackend::GetLayerSupport() const std::unique_ptr<IWorkloadFactory> NeonBackend::CreateWorkloadFactory() const { - // TODO implement return nullptr; } +void NeonBackend::Destroy(IBackend* backend) +{ + delete boost::polymorphic_downcast<NeonBackend*>(backend); +} + } // namespace armnn
\ No newline at end of file diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 86fccf0e51..5d4bd5dfcc 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -4,8 +4,7 @@ // #pragma once -#include "backends/IBackendInternal.hpp" - +#include <backends/IBackendInternal.hpp> #include "NeonLayerSupport.hpp" namespace armnn @@ -23,10 +22,9 @@ public: std::unique_ptr<IWorkloadFactory> CreateWorkloadFactory() const override; -private: - static const std::string s_Id; + static void Destroy(IBackend* backend); - // TODO initialize +private: NeonLayerSupport m_LayerSupport; }; |