moco::tf::TFOptimizer Class Reference

#include <TFOptimizer.h>

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

Detailed Description

Definition at line 27 of file TFOptimizer.h.

Member Function Documentation

optimize()

void moco::tf::TFOptimizer::optimize

(

loco::Graph *

g

)

const

Definition at line 32 of file TFOptimizer.cpp.

{
  logo::Phase phase;
 
  /* TRANSFORM DECLARATION BEGIN */
  if (moco::tf::get<moco::tf::Knob::ResolveFusedBatchNorm>())
  {
    phase.emplace_back(std::make_unique<moco::ResolveFusedBatchNorm>());
  }
  if (moco::tf::get<moco::tf::Knob::FuseBinaryIntoPreceding>())
  {
    phase.emplace_back(std::make_unique<moco::FuseBinaryIntoPreceding>());
  }
  if (moco::tf::get<moco::tf::Knob::ResolveConstantShape>())
  {
    phase.emplace_back(std::make_unique<moco::ResolveConstantShape>());
  }
  if (moco::tf::get<moco::tf::Knob::ResolveReshapeWildcardDim>())
  {
    phase.emplace_back(std::make_unique<moco::ResolveReshapeWildcardDim>());
  }
  if (moco::tf::get<moco::tf::Knob::ResolveSquaredDifference>())
  {
    phase.emplace_back(std::make_unique<moco::ResolveSquaredDifference>());
  }
  if (moco::tf::get<moco::tf::Knob::RemoveTFIdentityNode>())
  {
    phase.emplace_back(std::make_unique<moco::RemoveTFIdentityNode>());
  }
  if (moco::tf::get<moco::tf::Knob::RemoveDeadNode>())
  {
    phase.emplace_back(std::make_unique<logo::RemoveDeadNodePass>());
  }
  if (moco::tf::get<moco::tf::Knob::SqueezeReduceNode>())
  {
    phase.emplace_back(std::make_unique<moco::SqueezeReduceNode>());
  }
  // Shape inference is needed for added nodes doing above transformations
  phase.emplace_back(std::make_unique<moco::tf::ShapeInferencePass>());
  phase.emplace_back(std::make_unique<moco::tf::TypeInferencePass>());
  /* 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/TFOptimizer.h
• compiler/moco-tf/src/TFOptimizer.cpp