From 1a26896fd8d48205393ba0f22db864b5302b703f Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Mon, 27 Jul 2020 15:52:13 +0100
Subject: IVGCVSW-5011 Implement GetCapabilities in NeonTensorHandleFactory

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I43591ec1250c1d84d286de85956a86eb5e2abc2a
---
 src/backends/neon/NeonTensorHandleFactory.hpp | 28 +++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

(limited to 'src/backends/neon/NeonTensorHandleFactory.hpp')

diff --git a/src/backends/neon/NeonTensorHandleFactory.hpp b/src/backends/neon/NeonTensorHandleFactory.hpp
index e1cdc8bbac..0930d4e8d7 100644
--- a/src/backends/neon/NeonTensorHandleFactory.hpp
+++ b/src/backends/neon/NeonTensorHandleFactory.hpp
@@ -13,6 +13,30 @@ namespace armnn
 
 constexpr const char* NeonTensorHandleFactoryId() { return "Arm/Neon/TensorHandleFactory"; }
 
+const std::set<armnn::LayerType> paddingRequiredLayers {
+    LayerType::ArgMinMax,
+    LayerType::Concat,
+    LayerType::Convolution2d,
+    LayerType::DepthToSpace,
+    LayerType::DepthwiseConvolution2d,
+    LayerType::Dequantize,
+    LayerType::FullyConnected,
+    LayerType::Gather,
+    LayerType::L2Normalization,
+    LayerType::Lstm,
+    LayerType::Mean,
+    LayerType::Multiplication,
+    LayerType::Normalization,
+    LayerType::Permute,
+    LayerType::Pooling2d,
+    LayerType::Quantize,
+    LayerType::QuantizedLstm,
+    LayerType::Resize,
+    LayerType::Stack,
+    LayerType::Transpose,
+    LayerType::TransposeConvolution2d
+};
+
 class NeonTensorHandleFactory : public ITensorHandleFactory
 {
 public:
@@ -46,6 +70,10 @@ public:
 
     MemorySourceFlags GetImportFlags() const override;
 
+    std::vector<Capability> GetCapabilities(const IConnectableLayer* layer,
+                                            const IConnectableLayer* connectedLayer,
+                                            CapabilityClass capabilityClass) override;
+
 private:
     mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager;
 };
-- 
cgit v1.2.1