[moco-tf] Enable ResolveReshapeWildcardDim transform (#6154)

This commit registers and enables ResolveReshapeWildcardDim 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 9001948..baf3deb 100644 (file)
--- a/compiler/moco-tf/src/TFOptimizer.cpp
+++ b/compiler/moco-tf/src/TFOptimizer.cpp
@@ -44,6 +44,7 @@ void TFOptimizer::optimize(loco::Graph *g) const
   // 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::FixPaddingTransform>());
+  phase.emplace_back(stdex::make_unique<moco::tf::ResolveReshapeWildcardDim>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};

diff --git a/compiler/moco-tf/src/Transforms.h b/compiler/moco-tf/src/Transforms.h
index 3956245..accde2b 100644 (file)
--- a/compiler/moco-tf/src/Transforms.h
+++ b/compiler/moco-tf/src/Transforms.h
@@ -26,6 +26,7 @@
 #include "Transforms/RemoveForwardNodeTransform.h"
 #include "Transforms/ReorderDecodeTransform.h"
 #include "Transforms/ResolveFusedBatchNorm.h"
+#include "Transforms/ResolveReshapeWildcardDim.h"
 #include "Transforms/SimplifyDomainConversionTransform.h"
 
 #endif // __MOCO_TF_TRANSFORMS_H__