From 8359c28329c2684cad1603dae0c7f7f87561adb2 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Thu, 11 Oct 2018 13:56:25 +0900 Subject: [PATCH] [enco] Generate Pad Op tree for AvgPool lowering (#1829) This commit revises AvgPool lowering pass to generate Eval instruction with Pad Op tree (instead of UnitF). Signed-off-by: Jonghyun Park --- contrib/enco/core/src/Transforms/AvgPoolLowering.cpp | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/contrib/enco/core/src/Transforms/AvgPoolLowering.cpp b/contrib/enco/core/src/Transforms/AvgPoolLowering.cpp index cb20aeb..43a1387 100644 --- a/contrib/enco/core/src/Transforms/AvgPoolLowering.cpp +++ b/contrib/enco/core/src/Transforms/AvgPoolLowering.cpp @@ -92,17 +92,22 @@ public: auto m = ifm_obj->module(); assert(m != nullptr); + auto load_op = m->entity()->op()->create(); + + load_op->object(ifm_obj); + auto pad_op = m->entity()->op()->create(); + pad_op->arg(load_op); + pad_op->pad()->top(_pad->top()); pad_op->pad()->bottom(_pad->bottom()); pad_op->pad()->left(_pad->left()); pad_op->pad()->right(_pad->right()); - auto pad_instr = m->entity()->instr()->create(); + auto pad_instr = m->entity()->instr()->create(); - pad_instr->ifm(ifm_obj); - pad_instr->ofm(ofm_obj); + pad_instr->out(ofm_obj); pad_instr->op(pad_op); return pad_instr; -- 2.7.4