#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
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>())
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);
}