[moco-tf] Enable Squeeze Canonicalizer (#6261)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Tue, 6 Aug 2019 03:39:05 +0000 (12:39 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Tue, 6 Aug 2019 03:39:05 +0000 (12:39 +0900)
This commit register and enables SqueezeCanonicalizer

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
compiler/moco-tf/src/Canonicalizer.cpp

index ba1c4e7..accd955 100644 (file)
@@ -32,6 +32,7 @@
 #include "Canonicalization/ReluCanonicalizer.h"
 #include "Canonicalization/Relu6Canonicalizer.h"
 #include "Canonicalization/ReshapeCanonicalizer.h"
+#include "Canonicalization/SqueezeCanonicalizer.h"
 
 #include "Dialect/TFDialect.h"
 #include "Dialect/TFNodes.h"
@@ -83,6 +84,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<ReluCanonicalizer>());
   phase.emplace_back(stdex::make_unique<Relu6Canonicalizer>());
   phase.emplace_back(stdex::make_unique<ReshapeCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<SqueezeCanonicalizer>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};