From fc8ece53fceef128a679a378ab1936caf04ce301 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?=
 =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <saehie.park@samsung.com>
Date: Wed, 21 Aug 2019 14:37:51 +0900
Subject: [PATCH] [moco-tf] RemoveDeadNode for TF dialects (#6773)

This will add RemoveDeadNode optimization for TF dialect node transforms

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---
 compiler/moco-tf/src/TFOptimizer.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/compiler/moco-tf/src/TFOptimizer.cpp b/compiler/moco-tf/src/TFOptimizer.cpp
index 7e04c21..eb5476b 100644
--- a/compiler/moco-tf/src/TFOptimizer.cpp
+++ b/compiler/moco-tf/src/TFOptimizer.cpp
@@ -48,6 +48,10 @@ void TFOptimizer::optimize(loco::Graph *g) const
   {
     phase.emplace_back(stdex::make_unique<moco::tf::ResolveReshapeWildcardDim>());
   }
+  if (moco::tf::get<moco::tf::Knob::RemoveDeadNode>())
+  {
+    phase.emplace_back(stdex::make_unique<RemoveDeadNodeTransform>());
+  }
   // Fix shape and pad for added nodes doing above transformations
   // TODO need to merge or remove the ones in importer
   phase.emplace_back(stdex::make_unique<moco::tf::FixShapeTransform>());
-- 
2.7.4