Enabling Tanh on pureACL NEON runtime (#2674)
authorPrasanna R/System SW /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Wed, 12 Sep 2018 08:47:05 +0000 (14:17 +0530)
committer이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Wed, 12 Sep 2018 08:47:04 +0000 (17:47 +0900)
This patch enables the TANH operation on pureACL NEON runtime.

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

index 0a8c2e2..45a48ba 100644 (file)
@@ -2926,7 +2926,13 @@ void Planner::visit(const ::internal::tflite::op::Tanh::Node &node)
       builder.append("Tanh", std::move(fn));
     }
     else
-      throw std::runtime_error("Not supported, yet");
+    {
+      auto fn = nnfw::make_unique<::arm_compute::NEActivationLayer>();
+
+      fn->configure(ifm_alloc, ofm_alloc, act_info);
+
+      builder.append("Tanh", std::move(fn));
+    }
   };
 
   _builder.addStage(stage);