locop Namespace Reference

Data Structures
class	ArgDesc
class	CanonicalNodeSummaryBuilder
	Built-in Node Summary Builder for Canonical Dialect. More...
struct	FormattedGraph
class	FormattedGraphImpl
class	FormattedGraphImpl< Formatter::LinearV1 >
class	FormattedTensorShape
class	FormattedTensorShape< TensorShapeFormat::Bracket >
class	FormattedTensorShape< TensorShapeFormat::Plain >
class	GenericNodeSummaryBuilder
	Dialect-agnostic Node Summary Builder. More...
struct	NodeDesc
struct	NodeSummaryBuilder
	Build a summary from loco Node. More...
struct	NodeSummaryBuilderFactory
struct	Spec
struct	Spec< Interface::Formatted >
struct	SymbolTable
	Symbol Table Interface. More...

Typedefs
using	OpName = std::string
using	ArgName = std::string
using	ArgValue = std::string
using	ArgElem = std::pair< ArgName, ArgValue >
using	NodeSummary = NodeDesc

Enumerations
enum	Formatter { LinearV1 }
enum class	TensorShapeFormat { Plain , Bracket }
enum class	Interface { Formatted }

Functions
std::ostream &	operator<< (std::ostream &, const FormattedGraph &)
template<Formatter F>
FormattedGraphImpl< F >	fmt (loco::Graph *g)
template<Formatter F>
FormattedGraphImpl< F >	fmt (const std::unique_ptr< loco::Graph > &g)
template<TensorShapeFormat F>
FormattedTensorShape< F >	fmt (loco::TensorShape *ptr)
std::ostream &	operator<< (std::ostream &, const Spec< Interface::Formatted > &)
std::ostream &	operator<< (std::ostream &os, const NodeDesc &d)

Typedef Documentation

ArgElem

using locop::ArgElem = typedef std::pair<ArgName, ArgValue>

Definition at line 31 of file NodeSummary.h.

ArgName

using locop::ArgName = typedef std::string

Definition at line 29 of file NodeSummary.h.

ArgValue

using locop::ArgValue = typedef std::string

Definition at line 30 of file NodeSummary.h.

NodeSummary

using locop::NodeSummary = typedef NodeDesc

Definition at line 110 of file NodeSummary.h.

OpName

using locop::OpName = typedef std::string

Definition at line 28 of file NodeSummary.h.

Enumeration Type Documentation

Formatter

enum locop::Formatter

Enumerator
LinearV1

Definition at line 45 of file FormattedGraph.h.

{
  LinearV1,
  // TO BE ADDED
};

Interface

enum class locop::Interface

strong

Enumerator
Formatted

Definition at line 25 of file Interfaces.h.

{
  Formatted,
};

TensorShapeFormat

enum class locop::TensorShapeFormat

strong

Enumerator
Plain
Bracket

Definition at line 27 of file FormattedTensorShape.h.

{
  // D_0 x D_1 x ... D_N
  Plain,
  // [ D_0 x D_1 x D_2 x ... ]
  Bracket,
};

Function Documentation

fmt() [1/3]

template<Formatter F>

FormattedGraphImpl< F > locop::fmt

(

const std::unique_ptr< loco::Graph > &

g

)

Definition at line 82 of file FormattedGraph.h.

{
  return fmt<F>(g.get());
}

fmt() [2/3]

template<Formatter F>

FormattedGraphImpl< F > locop::fmt

(

loco::Graph *

g

)

Definition at line 77 of file FormattedGraph.h.

{
  return FormattedGraphImpl<F>{g};
}

Referenced by operator<<().

fmt() [3/3]

template<TensorShapeFormat F>

FormattedTensorShape< F > locop::fmt

(

loco::TensorShape *

ptr

)

Definition at line 69 of file FormattedTensorShape.h.

{
  return FormattedTensorShape<F>{ptr};
}

operator<<() [1/3]

std::ostream & locop::operator<<	(	std::ostream &	os,
		const FormattedGraph &	fmt
	)

Definition at line 207 of file FormattedGraph.cpp.

{
  fmt.dump(os);
  return os;
}

References fmt().

operator<<() [2/3]

std::ostream & locop::operator<<	(	std::ostream &	os,
		const Spec< Interface::Formatted > &	formatted
	)

Definition at line 22 of file Interfaces.cpp.

{
  formatted.dump(os);
  return os;
}

operator<<() [3/3]

std::ostream & locop::operator<<	(	std::ostream &	os,
		const NodeDesc &	d
	)

Definition at line 171 of file FormattedGraph.cpp.

{
  assert(d.state() != NodeDesc::State::Invalid);
 
  std::vector<std::string> values;
 
  for (uint32_t n = 0; n < d.args().count(); ++n)
  {
    values.emplace_back(d.args().at(n).first + ": " + d.args().at(n).second);
  }
 
  if (d.state() == NodeDesc::State::PartiallyKnown)
  {
    values.emplace_back("...");
  }
 
  os << d.opname();
  os << "(";
  if (values.size() > 0)
  {
    os << values.at(0);
    for (uint32_t n = 1; n < values.size(); ++n)
    {
      os << ", " << values.at(n);
    }
  }
  os << ")";
 
  return os;
}

References locop::NodeDesc::Invalid, and locop::NodeDesc::PartiallyKnown.