Namespaces
namespace	eltwise

namespace	test

Data Structures
class	AddCanonicalizer
	Convert TFAdd to Canonical EltwiseAdd. More...

class	AvgPoolCanonicalizer
	Convert TFAvgPool to Canonical AvgPool2D. More...

class	BiasAddCanonicalizer
	Convert TFBiasAdd to Canonical BiasAdd. More...

class	BroadcastFunctor

class	Canonicalizer

class	ConcatV2Canonicalizer
	Convert TFConcatV2 to Canonical TensorConcat. More...

class	ConstCanonicalizer
	Convert TFConst to Canonical ConstGen. More...

class	Conv2DBackpropInputCanonicalizer
	Convert TFConv2DBackpropInput to Canonical TransposedConv2D. More...

class	Conv2DCanonicalizer
	Convert TFConv2D to Canonical Conv2D. More...

class	COpCallGraphBuilder
	GraphBuilder for COpCall node. More...

class	DepthwiseConv2dNativeCanonicalizer
	Convert TFDepthwiseConv2dNative to Canonical DepthwiseConv2D. More...

class	Frontend

class	IdentityCanonicalizer
	Convert TFIdentity to Canonical Forward. More...

struct	KnobTrait

class	MaximumCanonicalizer
	Convert TFMaximum to Canonical EltwiseMax. More...

class	MaxPoolCanonicalizer
	Convert TFMaxPool to Canonical MaxPool2D. More...

class	MeanCanonicalizer
	Canonicalize TF-dialect TFMean into canonical TensorReduce(Mean) node. More...

class	MocoNodeSummaryBuilder

class	MulCanonicalizer
	Convert TFMul to Canonical EltwiseMul. More...

class	Optimizer

class	PadCanonicalizer
	Convert TFPad to Canonical TensorConstantPad. More...

class	PlaceholderCanonicalizer
	Convert TFPlaceholder to Canonical Pull. More...

class	ProgressReporter

class	RealDivCanonicalizer
	Convert TFRealDiv to Canonical EltwiseDiv. More...

class	Relu6Canonicalizer
	Convert TFRelu6 to Canonical ReLU6. More...

class	ReluCanonicalizer
	Convert TFRelu to Canonical ReLU. More...

class	ReshapeCanonicalizer
	Convert TFReshape to Canonical Reshape. More...

class	RsqrtCanonicalizer
	Convert TFRsqrt to Canonical EltwiseDiv + EltwiseSqrt. More...

class	ShapeInferencePass
	Run shape inference to the graph. More...

struct	SimpleNodeTransform
	Per-Node Transform. More...

class	SoftmaxCanonicalizer
	Canonicalize TF-dialect TFSoftmax into canonical Softmax node. More...

class	SqrtCanonicalizer
	Convert TFsqrt to Canonical EltwiseSqrt. More...

class	SqueezeCanonicalizer
	Canonicalize TF-dialect TFSqueeze into canonical FixedReshape node. More...

class	StopGradientCanonicalizer
	Canonicalize TF-dialect TFStopGradient into canonical Forward node. More...

class	SubCanonicalizer
	Convert TFSub to Canonical EltwiseSub. More...

class	TanhCanonicalizer
	Convert TFTanh to Canonical Tanh. More...

class	TFNodeSummaryBuilderFactory

class	TFOptimizer

class	TFPushCanonicalizer
	Convert TFPush to Canonical Push. More...

class	TypeInferencePass
	Run type inference to the graph. More...

Typedefs
using	FormattedGraph = locop::FormattedGraphImpl< locop::Formatter::LinearV1 >

using	Transform = logo::Pass

Enumerations
enum class	Knob { KNOB_BOOL }

Functions
template<Knob K>
KnobTrait< K >::ValueType	get (void)

FormattedGraph	fmt (loco::Graph *g)

std::string	transform_name (const Transform *t)

template<typename DERIVED >
DERIVED *	as (loco::Node *node)

Typedef Documentation

◆ FormattedGraph

using moco::tf::FormattedGraph = typedef locop::FormattedGraphImpl<locop::Formatter::LinearV1>

Definition at line 64 of file LogHelper.h.

◆ Transform

using moco::tf::Transform = typedef logo::Pass

Note: Transform will be replaced by logo::Pass

Definition at line 35 of file Transform.h.

Enumeration Type Documentation

◆ Knob

enum class moco::tf::Knob

strong

Enumerator
KNOB_BOOL

Definition at line 25 of file Knob.h.

{
#define KNOB_BOOL(NAME, DEFAULT, DESC) NAME,
#include "Knob.lst"
#undef KNOB_BOOL
};

Function Documentation

◆ as()

template<typename DERIVED >

DERIVED * moco::tf::as ( loco::Node * node )

Definition at line 39 of file Transform.h.

39{ return dynamic_cast<DERIVED *>(node); }

◆ fmt()

FormattedGraph moco::tf::fmt ( loco::Graph * g )

Definition at line 75 of file LogHelper.cpp.

{
  auto node_summary_builder = std::make_unique<TFNodeSummaryBuilderFactory>();
  return std::move(locop::fmt<locop::LinearV1>(g).with(std::move(node_summary_builder)));
}

Referenced by moco::tf::ProgressReporter::notify(), and moco::tf::ProgressReporter::notify().

◆ get()

template<Knob K>

KnobTrait< K >::ValueType moco::tf::get ( void )

◆ transform_name()

std::string moco::tf::transform_name ( const Transform * t )

Definition at line 24 of file Transform.cpp.

{
  if (t->name() == nullptr)
  {
    return "(unknown)";
  }
 
  return t->name();
}

References logo::Pass::name().

Namespaces

Data Structures

Typedefs

Enumerations

Functions