[moco-tf] Use logo for TFOptimizer (#7121)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 4 Sep 2019 01:44:40 +0000 (10:44 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Wed, 4 Sep 2019 01:44:40 +0000 (10:44 +0900)
This will revise TFOptimizer to use logo Phase and ProgressReporter

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

index 22154cd..9aea14d 100644 (file)
 #include "TFOptimizer.h"
 
 #include "Knob.h"
-#include "Phase.h"
+#include "ProgressReporter.h"
 #include "Transforms.h"
 
+#include <logo/Phase.h>
+
 #include <stdex/Memory.h>
 
 namespace moco
@@ -29,7 +31,7 @@ namespace tf
 
 void TFOptimizer::optimize(loco::Graph *g) const
 {
-  moco::tf::Phase phase;
+  logo::Phase phase;
 
   /* TRANSFORM DECLARATION BEGIN */
   if (moco::tf::get<moco::tf::Knob::ResolveFusedBatchNorm>())
@@ -61,7 +63,9 @@ void TFOptimizer::optimize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<moco::tf::FixShapeTransform>());
   /* TRANSFORM DECLARATION END */
 
-  moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};
+  ProgressReporter prog(g, logo::PhaseStrategy::Saturate);
+  logo::PhaseRunner<logo::PhaseStrategy::Saturate> phase_runner{g};
+  phase_runner.attach(&prog);
   phase_runner.run(phase);
 }