[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 <hyunsik.yoon@samsung.com>

diff --git a/compiler/moco-tf/src/Knob.lst b/compiler/moco-tf/src/Knob.lst
index 73382be..578ec4a 100644 (file)
--- 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 (file)
--- 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<moco::tf::Knob::ConstantFolding>())
+  {
+    phase.emplace_back(stdex::make_unique<ConstantFoldingTransform>());
+  }
+
   if (moco::tf::get<moco::tf::Knob::RemoveDeadNode>())
   {
     phase.emplace_back(stdex::make_unique<RemoveDeadNodeTransform>());