Enable explicit Conv2D on pureacl NEON runtime (#2470)
author이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Mon, 27 Aug 2018 01:34:59 +0000 (10:34 +0900)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Mon, 27 Aug 2018 01:34:59 +0000 (10:34 +0900)
- Enable explicit Conv2D on pureacl NEON runtime

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

index 2673cd9..0990685 100644 (file)
@@ -1078,7 +1078,13 @@ void Planner::visit(const ::internal::tflite::op::Conv2D::Explicit::Node &node)
       builder.append("Conv2D", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      std::unique_ptr<::arm_compute::NEConvolutionLayer> fn{new ::arm_compute::NEConvolutionLayer};
+
+      fn->configure(ifm_alloc, ker_alloc, bias_alloc, ofm_alloc, conv_info);
+
+      builder.append("Conv2D", std::move(fn));
+    }
 
     ActivationBuilder{builder}.append(param.activation, ofm_alloc);
   };