[moco-tf] Enable SquaredDifference canonicalizer (#6629)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Fri, 16 Aug 2019 03:43:31 +0000 (12:43 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Fri, 16 Aug 2019 03:43:31 +0000 (12:43 +0900)
This will enable SquaredDifference canonicalizer

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
compiler/moco-tf/src/Canonicalizer.cpp

index d91181d..f33a2f5 100644 (file)
@@ -35,6 +35,7 @@
 #include "Canonicalization/Relu6Canonicalizer.h"
 #include "Canonicalization/ReshapeCanonicalizer.h"
 #include "Canonicalization/SqrtCanonicalizer.h"
+#include "Canonicalization/SquaredDifferenceCanonicalizer.h"
 #include "Canonicalization/SqueezeCanonicalizer.h"
 #include "Canonicalization/SubCanonicalizer.h"
 
@@ -91,6 +92,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<Relu6Canonicalizer>());
   phase.emplace_back(stdex::make_unique<ReshapeCanonicalizer>());
   phase.emplace_back(stdex::make_unique<SqrtCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<SquaredDifferenceCanonicalizer>());
   phase.emplace_back(stdex::make_unique<SqueezeCanonicalizer>());
   phase.emplace_back(stdex::make_unique<SubCanonicalizer>());
   /* TRANSFORM DECLARATION END */