Enable Explicit MaxPool2d on pureacl NEON runtime (#2468)
author이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Fri, 24 Aug 2018 09:06:06 +0000 (18:06 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Fri, 24 Aug 2018 09:06:06 +0000 (18:06 +0900)
- Enable Explicit MaxPool2d on pureacl NEON runtime

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>
runtimes/pure_arm_compute/src/compilation.cc

index 5fe20b3..e1c7354 100644 (file)
@@ -1627,7 +1627,13 @@ void Planner::visit(const ::internal::tflite::op::MaxPool2D::Explicit::Node &nod
       builder.append("MaxPool2D", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      std::unique_ptr<::arm_compute::NEPoolingLayer> fn{new ::arm_compute::NEPoolingLayer};
+
+      fn->configure(ifm_alloc, ofm_alloc, info);
+
+      builder.append("MaxPool2D", std::move(fn));
+    }
 
     ActivationBuilder{builder}.append(param.activation, ofm_alloc);
   };