From 5ef25612df2d93ac4078e6bc2cdb65a6373161c8 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 14:28:04 +0900 Subject: [PATCH] [enco.caffe] Emit Eval instruction for ReLU (#1832) With this commit, caffe frontend now emits Eval instruction (instead of UnitF) for ReLU layers. Signed-off-by: Jonghyun Park --- contrib/enco/frontend/caffe/src/Frontend.cpp | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/contrib/enco/frontend/caffe/src/Frontend.cpp b/contrib/enco/frontend/caffe/src/Frontend.cpp index 31e1793..12907fa 100644 --- a/contrib/enco/frontend/caffe/src/Frontend.cpp +++ b/contrib/enco/frontend/caffe/src/Frontend.cpp @@ -419,15 +419,20 @@ enco::Bundle Frontend::load(void) const ofm_obj->bag(ofm_bag); ofm_obj->layout(coco::FeatureLayouts::BCHW::create(ofm_obj->shape())); + // Create a Load Op + auto load = m->entity()->op()->create(); + + load->object(ifm_obj); + // Create a ReLU op auto op = m->entity()->op()->create(); - // Create a UnitF instruction - // TODO Use UnitT later - auto ins = m->entity()->instr()->create(); + op->arg(load); - ins->ifm(ifm_obj); - ins->ofm(ofm_obj); + // Create a Eval instruction + auto ins = m->entity()->instr()->create(); + + ins->out(ofm_obj); ins->op(op); // Append the instruction to the block -- 2.7.4