moco::tf::Canonicalizer Class Reference

#include <Canonicalizer.h>

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

Detailed Description

Definition at line 27 of file Canonicalizer.h.

Member Function Documentation

canonicalize()

void moco::tf::Canonicalizer::canonicalize

(

loco::Graph *

g

)

const

Definition at line 88 of file Canonicalizer.cpp.

{
  logo::Phase phase;
 
  /* TRANSFORM DECLARATION BEGIN */
  // Run shape and type inference at the top
  phase.emplace_back(std::make_unique<ShapeInferencePass>());
  phase.emplace_back(std::make_unique<TypeInferencePass>());
 
  phase.emplace_back(std::make_unique<AddCanonicalizer>());
  phase.emplace_back(std::make_unique<AvgPoolCanonicalizer>());
  if (moco::tf::get<moco::tf::Knob::CanonicalizeBiasAdd>())
    phase.emplace_back(std::make_unique<BiasAddCanonicalizer>());
  phase.emplace_back(std::make_unique<ConcatV2Canonicalizer>());
  if (moco::tf::get<moco::tf::Knob::CanonicalizeConst>())
    phase.emplace_back(std::make_unique<ConstCanonicalizer>());
  phase.emplace_back(std::make_unique<Conv2DBackpropInputCanonicalizer>());
  if (moco::tf::get<moco::tf::Knob::CanonicalizeConv2D>())
    phase.emplace_back(std::make_unique<Conv2DCanonicalizer>());
  phase.emplace_back(std::make_unique<DepthwiseConv2dNativeCanonicalizer>());
  phase.emplace_back(std::make_unique<IdentityCanonicalizer>());
  phase.emplace_back(std::make_unique<MaximumCanonicalizer>());
  phase.emplace_back(std::make_unique<MaxPoolCanonicalizer>());
  phase.emplace_back(std::make_unique<MeanCanonicalizer>());
  phase.emplace_back(std::make_unique<MulCanonicalizer>());
  phase.emplace_back(std::make_unique<PadCanonicalizer>());
  phase.emplace_back(std::make_unique<PlaceholderCanonicalizer>());
  phase.emplace_back(std::make_unique<RealDivCanonicalizer>());
  phase.emplace_back(std::make_unique<ReluCanonicalizer>());
  phase.emplace_back(std::make_unique<Relu6Canonicalizer>());
  phase.emplace_back(std::make_unique<ReshapeCanonicalizer>());
  phase.emplace_back(std::make_unique<RsqrtCanonicalizer>());
  phase.emplace_back(std::make_unique<SoftmaxCanonicalizer>());
  phase.emplace_back(std::make_unique<SqrtCanonicalizer>());
  // NOTE SquaredDifference is handled in ResolveSquaredDifference
  phase.emplace_back(std::make_unique<SqueezeCanonicalizer>());
  phase.emplace_back(std::make_unique<StopGradientCanonicalizer>());
  phase.emplace_back(std::make_unique<SubCanonicalizer>());
  phase.emplace_back(std::make_unique<TanhCanonicalizer>());
  // For virtual nodes
  phase.emplace_back(std::make_unique<TFPushCanonicalizer>());
  /* TRANSFORM DECLARATION END */
 
  ProgressReporter prog(g, logo::PhaseStrategy::Restart);
  logo::PhaseRunner<logo::PhaseStrategy::Restart> phase_runner{g};
  phase_runner.attach(&prog);
  phase_runner.run(phase);
 
  // Assert if graph has TF dialect nodes
  assert(!has_tf_nodes(g));
}

References logo::Restart.

---

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

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