[moco-tf] Enable ResolveConstantShape (#6423)

This commit enables ResolveConstantShape transform

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

diff --git a/compiler/moco-tf/src/TFOptimizer.cpp b/compiler/moco-tf/src/TFOptimizer.cpp
index 4789df2..d295756 100644 (file)
--- a/compiler/moco-tf/src/TFOptimizer.cpp
+++ b/compiler/moco-tf/src/TFOptimizer.cpp
@@ -43,6 +43,7 @@ void TFOptimizer::optimize(loco::Graph *g) const
   // 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>());
+  phase.emplace_back(stdex::make_unique<moco::tf::ResolveConstantShape>());
   phase.emplace_back(stdex::make_unique<moco::tf::ResolveReshapeWildcardDim>());
   /* TRANSFORM DECLARATION END */
 

diff --git a/compiler/moco-tf/src/Transforms.h b/compiler/moco-tf/src/Transforms.h
index cfc607d..b108688 100644 (file)
--- a/compiler/moco-tf/src/Transforms.h
+++ b/compiler/moco-tf/src/Transforms.h
@@ -24,6 +24,7 @@
 #include "Transforms/RemoveDeadNodeTransform.h"
 #include "Transforms/RemoveForwardNodeTransform.h"
 #include "Transforms/ReorderDecodeTransform.h"
+#include "Transforms/ResolveConstantShape.h"
 #include "Transforms/ResolveFusedBatchNorm.h"
 #include "Transforms/ResolveReshapeWildcardDim.h"
 #include "Transforms/SimplifyDomainConversionTransform.h"