From 5662c206864df4121eea29c541c24c0f62113809 Mon Sep 17 00:00:00 2001
From: Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com>
Date: Mon, 24 Sep 2018 13:30:09 +0100
Subject: IVGCVSW-1881 Add PadLayer to ArmNN

Change-Id: Ib70883a20fa7b82bfb090aa98cc51a6d645b5722
---
 src/backends/WorkloadFactory.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/backends/WorkloadFactory.cpp')

diff --git a/src/backends/WorkloadFactory.cpp b/src/backends/WorkloadFactory.cpp
index 773a8c1a18..2666e7f354 100644
--- a/src/backends/WorkloadFactory.cpp
+++ b/src/backends/WorkloadFactory.cpp
@@ -472,6 +472,19 @@ bool IWorkloadFactory::IsLayerSupported(Compute compute, const Layer& layer, boo
                                         cLayer->GetParameters(), reason, reasonCapacity);
             break;
         }
+        case LayerType::Pad:
+        {
+            auto cLayer = boost::polymorphic_downcast<const PadLayer*>(&layer);
+            const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
+            const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
+            result = IsPadSupported(compute,
+                                    OverrideDataType(input, dataType),
+                                    OverrideDataType(output, dataType),
+                                    cLayer->GetParameters(),
+                                    reason,
+                                    reasonCapacity);
+            break;
+        }
         case LayerType::Pooling2d:
         {
             auto cLayer = boost::polymorphic_downcast<const Pooling2dLayer*>(&layer);
-- 
cgit v1.2.1