From 2e967e93ff1a02b7861457f4eca4fb8c2c80ecdc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= <liyinhgqw@gmail.com>
Date: Tue, 3 Apr 2018 12:36:13 +0800
Subject: [PATCH] Enable neon addn for mobilenet v2

---
 mace/kernels/addn.h | 6 ------
 mace/ops/addn.cc    | 6 ++++++
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/mace/kernels/addn.h b/mace/kernels/addn.h
index c8bb6016..61e906bf 100644
--- a/mace/kernels/addn.h
+++ b/mace/kernels/addn.h
@@ -77,12 +77,6 @@ struct AddNFunctor {
   }
 };
 
-template <>
-void AddNFunctor<DeviceType::NEON, float>::operator()(
-    const std::vector<const Tensor *> &input_tensors,
-    Tensor *output_tensor,
-    StatsFuture *future);
-
 template <typename T>
 struct AddNFunctor<DeviceType::OPENCL, T> {
   void operator()(const std::vector<const Tensor *> &input_tensors,
diff --git a/mace/ops/addn.cc b/mace/ops/addn.cc
index 8e253cf1..e30275bb 100644
--- a/mace/ops/addn.cc
+++ b/mace/ops/addn.cc
@@ -25,6 +25,12 @@ void Register_AddN(OperatorRegistry *op_registry) {
                                      .TypeConstraint<half>("T")
                                      .Build(),
                     AddNOp<DeviceType::OPENCL, half>);
+
+  REGISTER_OPERATOR(op_registry, OpKeyBuilder("AddN")
+                                     .Device(DeviceType::NEON)
+                                     .TypeConstraint<float>("T")
+                                     .Build(),
+                    AddNOp<DeviceType::NEON, float>);
 }
 
 }  // namespace ops
-- 
GitLab