[enco] Build ANN IR from ANNConv2D instruction (#2676)

This commit extends Split pass to build ANN IR from ANNConv2D
instruction.

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 40d7b7d..7d55a47 100644 (file)
--- a/contrib/enco/core/src/Transforms/Split.cpp
+++ b/contrib/enco/core/src/Transforms/Split.cpp
@@ -866,6 +866,25 @@ std::unique_ptr<ANNOpAppender> make_appender(coco::Instr *ins)
     return std::move(app);
   }
 
+  // Build ANN IR from ANNConv2D instruction
+  if (auto conv2d = coco::safe_cast<ANNConv2D>(ins))
+  {
+    auto sess = enco::session(conv2d->module());
+    auto app = make_unique<ANNConv2DAppender>();
+
+    app->session(sess);
+
+    app->pad(conv2d->pad());
+    app->stride(conv2d->stride());
+
+    app->ofm(conv2d->ofm()->asFeature());
+    app->ifm(conv2d->ifm()->asFeature());
+    app->ker(conv2d->ker()->asKernel());
+    app->bias(coco::safe_cast<coco::FeatureObject>(conv2d->bias()));
+
+    return std::move(app);
+  }
+
   return nullptr;
 }