moco::tf::Optimizer Class Reference

#include <Optimizer.h>

Public Member Functions
void	optimize (loco::Graph *) const

Detailed Description

Definition at line 27 of file Optimizer.h.

Member Function Documentation

optimize()

void moco::tf::Optimizer::optimize

(

loco::Graph *

g

)

const

Definition at line 32 of file Optimizer.cpp.

{
  logo::Phase phase;
 
  /* TRANSFORM DECLARATION BEGIN */
  // Shape inference is required for ResolveRedundantReshape
  phase.emplace_back(std::make_unique<ShapeInferencePass>());
 
  if (moco::tf::get<moco::tf::Knob::ConstantFolding>())
  {
    phase.emplace_back(std::make_unique<logo::ConstantFoldingPass>());
  }
 
  if (moco::tf::get<moco::tf::Knob::RemoveDeadNode>())
  {
    phase.emplace_back(std::make_unique<logo::RemoveDeadNodePass>());
  }
 
  if (moco::tf::get<moco::tf::Knob::ReorderDecode>() &&
      moco::tf::get<moco::tf::Knob::ReorderDecodeTensorBiasAdd>())
  {
    phase.emplace_back(std::make_unique<logo::ReorderDecodePass<loco::TensorBiasAdd>>());
  }
 
  if (moco::tf::get<moco::tf::Knob::ReorderDecode>() &&
      moco::tf::get<moco::tf::Knob::ReorderDecodeReLU>())
  {
    phase.emplace_back(std::make_unique<logo::ReorderDecodePass<loco::ReLU>>());
  }
 
  if (moco::tf::get<moco::tf::Knob::SimplifyDomainConversion>())
  {
    phase.emplace_back(std::make_unique<logo::SimplifyDomainConversionPass>());
  }
 
  if (moco::tf::get<moco::tf::Knob::RemoveForwardNode>())
  {
    phase.emplace_back(std::make_unique<logo::RemoveForwardNodePass>());
  }
 
  if (moco::tf::get<moco::tf::Knob::ResolveDuplicateReshape>())
  {
    phase.emplace_back(std::make_unique<logo::ResolveDuplicateReshapePass>());
  }
 
  if (moco::tf::get<moco::tf::Knob::ResolveRedundantReshape>())
  {
    phase.emplace_back(std::make_unique<logo::ResolveRedundantReshapePass>());
  }
  /* TRANSFORM DECLARATION END */
 
  ProgressReporter prog(g, logo::PhaseStrategy::Saturate);
  logo::PhaseRunner<logo::PhaseStrategy::Saturate> phase_runner{g};
  phase_runner.attach(&prog);
  phase_runner.run(phase);
}

References logo::Saturate.

---

The documentation for this class was generated from the following files:

• compiler/moco-tf/src/Optimizer.h
• compiler/moco-tf/src/Optimizer.cpp