From 467efad091d1acf5211e25e72e94ec712a0b4002 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 1 Aug 2019 15:49:47 +0900 Subject: [PATCH] [moco-tf] adding constant folding into optimizer (#6065) This commit adds constant folding into optimizer and turns the Knob ON. Signed-off-by: Hyun Sik Yoon --- compiler/moco-tf/src/Knob.lst | 1 + compiler/moco-tf/src/Optimizer.cpp | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/compiler/moco-tf/src/Knob.lst b/compiler/moco-tf/src/Knob.lst index 73382be..578ec4a 100644 --- a/compiler/moco-tf/src/Knob.lst +++ b/compiler/moco-tf/src/Knob.lst @@ -22,6 +22,7 @@ KNOB_BOOL(CanonicalizeConst, true, Enable Canonicalize for Const node) KNOB_BOOL(CanonicalizeConv2D, true, Enable Canonicalize for Conv2D node) // Canonical transforms +KNOB_BOOL(ConstantFolding, true, Enable constant-folding optimization) KNOB_BOOL(RemoveForwardNode, true, Enable RemoveForwardNode optimization) KNOB_BOOL(ReorderDecode, true, Enable ReorderDecode optimization) // BEG: These knobs are valid only when ReorderDecode is enabled diff --git a/compiler/moco-tf/src/Optimizer.cpp b/compiler/moco-tf/src/Optimizer.cpp index 46bfbe2..819b40e 100644 --- a/compiler/moco-tf/src/Optimizer.cpp +++ b/compiler/moco-tf/src/Optimizer.cpp @@ -32,6 +32,11 @@ void Optimizer::optimize(loco::Graph *g) const moco::tf::Phase phase; /* TRANSFORM DECLARATION BEGIN */ + if (moco::tf::get()) + { + phase.emplace_back(stdex::make_unique()); + } + if (moco::tf::get()) { phase.emplace_back(stdex::make_unique()); -- 2.7.4