[moco-tf] Enable TFReshape canonicalizer (#6151)

This commit registers TFReshape canonicalizer to enable it.

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

diff --git a/compiler/moco-tf/src/Canonicalizer.cpp b/compiler/moco-tf/src/Canonicalizer.cpp
index 89d6d63..80c5fca 100644 (file)
--- a/compiler/moco-tf/src/Canonicalizer.cpp
+++ b/compiler/moco-tf/src/Canonicalizer.cpp
@@ -29,6 +29,7 @@
 #include "Canonicalization/IdentityCanonicalizer.h"
 #include "Canonicalization/MaxPoolCanonicalizer.h"
 #include "Canonicalization/MulCanonicalizer.h"
+#include "Canonicalization/ReshapeCanonicalizer.h"
 
 #include "Dialect/TFDialect.h"
 #include "Dialect/TFNodes.h"
@@ -77,6 +78,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<IdentityCanonicalizer>());
   phase.emplace_back(stdex::make_unique<MaxPoolCanonicalizer>());
   phase.emplace_back(stdex::make_unique<MulCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<ReshapeCanonicalizer>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};