diff options
| author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-04 10:44:24 +0900 |
|---|---|---|
| committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-09-04 10:44:24 +0900 |
| commit | 7a0cfa5b230cafceccd8e941e7a410e94d4669de (patch) | |
| tree | be962632e05b5c1fdaffc12cd20ade7f008bed3b /compiler/moco-tf | |
| parent | f6f42e9555da4475f6d9c6995eb12abf15370e7a (diff) | |
| download | nnfw-7a0cfa5b230cafceccd8e941e7a410e94d4669de.tar.gz nnfw-7a0cfa5b230cafceccd8e941e7a410e94d4669de.tar.bz2 nnfw-7a0cfa5b230cafceccd8e941e7a410e94d4669de.zip | |
[moco-tf] Use logo for Optimizer (#7120)
This will revise Optimizer to use logo Phase and ProgressReporter
Signed-off-by: SaeHie Park <saehie.park@samsung.com>
Diffstat (limited to 'compiler/moco-tf')
| -rw-r--r-- | compiler/moco-tf/src/Optimizer.cpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/compiler/moco-tf/src/Optimizer.cpp b/compiler/moco-tf/src/Optimizer.cpp index fee4d020b..f33b4109b 100644 --- a/compiler/moco-tf/src/Optimizer.cpp +++ b/compiler/moco-tf/src/Optimizer.cpp @@ -17,9 +17,11 @@ #include "Optimizer.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 Optimizer::optimize(loco::Graph *g) const { - moco::tf::Phase phase; + logo::Phase phase; /* TRANSFORM DECLARATION BEGIN */ // Shape inference is required for ResolveRedundantReshape @@ -78,7 +80,9 @@ void Optimizer::optimize(loco::Graph *g) const } /* 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); } |