Enable FLOOR on pureacl NEON runtime (#2664)
authorPrasanna R/System SW /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Tue, 11 Sep 2018 08:30:42 +0000 (14:00 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 11 Sep 2018 08:30:42 +0000 (17:30 +0900)
This patch enables the ```Floor``` operation on pureACL ```NEON``` runtime.

Signed-off-by: prasannar <prasanna.r@samusng.com>
runtimes/pure_arm_compute/src/compilation.cc

index 74ca80d..a211778 100644 (file)
@@ -51,6 +51,7 @@
 #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 <arm_compute/runtime/NEON/functions/NEFloor.h>
 
 #include "internal/arm_compute.h"
 #include "internal/arm_compute/Cast.h"
@@ -3265,7 +3266,13 @@ void Planner::visit(const ::internal::tflite::op::Floor::Node &node)
       builder.append("Floor", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      auto fn = nnfw::make_unique<::arm_compute::NEFloor>();
+
+      fn->configure(ifm_alloc, ofm_alloc);
+
+      builder.append("Floor", std::move(fn));
+    }
   };
 
   _builder.addStage(stage);