From bd82d89652fb2935696bdf875a9c2d1cb06fe0cb Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 6 Aug 2019 12:39:05 +0900 Subject: [PATCH] [moco-tf] Enable Squeeze Canonicalizer (#6261) This commit register and enables SqueezeCanonicalizer Signed-off-by: Cheongyo Bahk --- compiler/moco-tf/src/Canonicalizer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/compiler/moco-tf/src/Canonicalizer.cpp b/compiler/moco-tf/src/Canonicalizer.cpp index ba1c4e7..accd955 100644 --- a/compiler/moco-tf/src/Canonicalizer.cpp +++ b/compiler/moco-tf/src/Canonicalizer.cpp @@ -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()); phase.emplace_back(stdex::make_unique()); phase.emplace_back(stdex::make_unique()); + phase.emplace_back(stdex::make_unique()); /* TRANSFORM DECLARATION END */ moco::tf::PhaseRunner phase_runner{g}; -- 2.7.4