[moco/tf] Enable Canonicalizer (#4148)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Tue, 9 Jul 2019 01:34:59 +0000 (10:34 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Tue, 9 Jul 2019 01:34:59 +0000 (10:34 +0900)
This will enable two existing two canonicalizer in transformation phase

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

index 40d19f9..5938f3d 100644 (file)
@@ -20,6 +20,9 @@
 #include "Phase.h"
 #include "Transforms.h"
 
+#include "Canonicalization/BiasAddCanonicalizer.h"
+#include "Canonicalization/Conv2DCanonicalizer.h"
+
 #include <stdex/Memory.h>
 
 namespace moco
@@ -32,7 +35,8 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   moco::tf::Phase phase;
 
   /* TRANSFORM DECLARATION BEGIN */
-
+  phase.emplace_back(stdex::make_unique<BiasAddCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<Conv2DCanonicalizer>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};