IVGCVSW-3478 Refactor armnn to make new Lstm parameters available in android-nn-driver through ILayerSupport.h

!android-nn-driver:1506

Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: Ia9904bd355c72a84f07d0a26faf143a4d6aead9c

1 files changed, 27 insertions, 0 deletions

diff --git a/src/armnn/BackendHelper.cpp b/src/armnn/BackendHelper.cpp
new file mode 100644
index 0000000000..2514038863
--- /dev/null
+++ b/src/armnn/BackendHelper.cpp
@@ -0,0 +1,27 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include <armnn/BackendHelper.hpp>
+#include "backendsCommon/BackendRegistry.hpp"
+#include "backendsCommon/IBackendInternal.hpp"
+
+namespace armnn
+{
+
+std::shared_ptr<ILayerSupport> GetILayerSupportByBackendId(const armnn::BackendId& backend)
+{
+    BackendRegistry& backendRegistry = armnn::BackendRegistryInstance();
+
+    if (!backendRegistry.IsBackendRegistered(backend))
+    {
+        return nullptr;
+    }
+
+    auto factoryFunc = backendRegistry.GetFactory(backend);
+    auto backendObject = factoryFunc();
+    return backendObject->GetLayerSupport();
+}
+
+}