[enco] Build ANN ReLU operator (#1298)

This commit revises ANNGroupBuilder and ANNOpBuilder to recognize ReLU
operation.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/enco/core/src/Transforms/Split.cpp b/contrib/enco/core/src/Transforms/Split.cpp
index 95664e0..2236e00 100644 (file)
--- a/contrib/enco/core/src/Transforms/Split.cpp
+++ b/contrib/enco/core/src/Transforms/Split.cpp
@@ -58,6 +58,12 @@ Compatibility ANNGroupBuilder::kind(const coco::Instr *ins) const
       // TODO Check data layout
       return true;
     }
+
+    bool visit(const coco::ReLU *) override
+    {
+      // TODO Check data layout
+      return true;
+    }
   };
 
   return ins->accept(CompatibilityCheck{}) ? COMPATIBLE : INCOMPATIBLE;
@@ -225,6 +231,13 @@ public:
                             {ifm, ker, bias, left, right, top, bottom, hstride, vstride, fuse},
                             {ofm});
     }
+    else if (unit->op()->asReLU())
+    {
+      auto ifm = _binder->addOperand<float>(unit->ifm());
+      auto ofm = _binder->addOperand<float>(unit->ofm());
+
+      _binder->addOperation(ann::Operation::Code::RELU, {ifm}, {ofm});
+    }
     else
     {
       throw std::runtime_error{"Not supported, yet"};