[enco] Bugfix about ReLU6 for backend (#2273)
author남궁석/동작제어Lab(SR)/Engineer/삼성전자 <sk.namkoong@samsung.com>
Wed, 14 Nov 2018 08:19:29 +0000 (17:19 +0900)
committer박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Wed, 14 Nov 2018 08:19:29 +0000 (17:19 +0900)
There are some typos for ReLU6 operation
This commit will fix them

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
contrib/enco/core/src/Transforms/Split.cpp

index 178ffab..86881fa 100644 (file)
@@ -739,7 +739,7 @@ public:
     }
     else if (auto relu6 = eval->op()->asReLU6())
     {
-      if (auto load = relu->arg()->asLoad())
+      if (auto load = relu6->arg()->asLoad())
       {
         // Let's compile the following code fragment:
         //
@@ -751,7 +751,7 @@ public:
 
         assert(ifm != nullptr && ofm != nullptr);
 
-        auto app = make_unique<ANNReLUAppender>();
+        auto app = make_unique<ANNReLU6Appender>();
 
         app->ifm(ifm);
         app->ofm(ofm);