From af96ca3fd9ce2c4b8eee18f14c7dc704c1cde7c0 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <hyunsik.yoon@samsung.com>
Date: Wed, 5 Sep 2018 17:36:15 +0900
Subject: [PATCH] [PureACL] Enable NEON to run Mobilenet (#2600)

This commit enables NEON to run Mobilenet by providing `depthwise_conv` and `reshape` for NEON.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---
 runtimes/pure_arm_compute/src/compilation.cc | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc
index 1fdc23d..9145348 100644
--- a/runtimes/pure_arm_compute/src/compilation.cc
+++ b/runtimes/pure_arm_compute/src/compilation.cc
@@ -34,6 +34,7 @@
 #include <arm_compute/runtime/NEON/functions/NEPoolingLayer.h>
 #include <arm_compute/runtime/NEON/functions/NEActivationLayer.h>
 #include <arm_compute/runtime/NEON/functions/NEConvolutionLayer.h>
+#include <arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h>
 
 #include "internal/arm_compute.h"
 #include "internal/arm_compute/Cast.h"
@@ -1265,7 +1266,13 @@ void Planner::visit(const ::internal::tflite::op::DepthwiseConv2D::Implicit::Nod
       builder.append("DepthwiseConv2D", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      auto fn = nnfw::make_unique<::arm_compute::NEDepthwiseConvolutionLayer>();
+
+      fn->configure(ifm_alloc, ker_alloc, bias_alloc, ofm_alloc, conv_info, param.multipler);
+
+      builder.append("DepthwiseConv2D", std::move(fn));
+    }
 
     ActivationBuilder{builder}.append(param.activation, ofm_alloc);
   };
@@ -2192,7 +2199,13 @@ void Planner::visit(const ::internal::tflite::op::Reshape::Node &node)
       builder.append("Reshape", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      auto fn = nnfw::make_unique<GenericReshapeLayer>();
+
+      fn->configure(input_alloc, output_alloc);
+
+      builder.append("Reshape", std::move(fn));
+    }
   };
 
   _builder.addStage(stage);
-- 
2.7.4