From: 박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Date: Fri, 2 Aug 2019 08:10:03 +0000 (+0900)
Subject: [moco-tf] Enable ResolveReshapeWildcardDim transform (#6154)
X-Git-Tag: submit/tizen/20190809.050447~215
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=f71972df0c777797241dfa08aac3176bd356e30f;p=platform%2Fcore%2Fml%2Fnnfw.git

[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
--- 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
--- 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__