locomotiv Namespace Reference

Data Structures
struct	BinaryFunc
struct	NodeData
	Read-only no-template wrapper for 'Buffer'. Serves interface for input and output of 'Session'. More...
class	NodeDataImpl
	An implementation of NodeData interface. More...
struct	NodeDomain
class	NodeExecution
	Helper class for Session, responsible to process one node calculation. More...
class	Session
	Session for loco graph inference. More...
struct	UnaryFunc

Functions
template<typename DT >
std::unique_ptr< NodeData >	make_data (const NodeData::Buffer< DT > &buffer)
	Copy buffer to make NodeData.
nncc::core::ADT::tensor::Index	as_nncc_index (const loco::TensorIndex &index)
loco::TensorIndex	as_loco_index (const nncc::core::ADT::tensor::Index &index)
template<>
std::unique_ptr< NodeData >	make_data (const NodeData::Buffer< int32_t > &buf)
template<>
std::unique_ptr< NodeData >	make_data (const NodeData::Buffer< float > &buf)
void	annot_domain (loco::Node *node, const loco::Domain &domain)
	Wrapper to annotate domain to node. Cannot annotate unknown domain.
loco::Domain	annot_domain (const loco::Node *node)
	Wrapper to get domain annotation of node.
void	erase_annot_domain (loco::Node *node)
	Erase already annotated node domain.
void	validate (bool true_cond, const std::string &&exception_msg)

Function Documentation

annot_domain() [1/2]

loco::Domain locomotiv::annot_domain

(

const loco::Node *

node

)

Wrapper to get domain annotation of node.

Definition at line 42 of file NodeDomain.cpp.

{
  auto node_domain = node->annot<NodeDomain>();
  if (node_domain)
    return node_domain->value;
  else
    return loco::Domain::Unknown;
}

References loco::AnnotatedItem< Annotation >::annot(), loco::Unknown, and locomotiv::NodeDomain::value.

annot_domain() [2/2]

void locomotiv::annot_domain	(	loco::Node *	node,
		const loco::Domain &	domain
	)

Wrapper to annotate domain to node. Cannot annotate unknown domain.

Definition at line 34 of file NodeDomain.cpp.

{
  assert(domain != loco::Domain::Unknown);
  auto node_domain = std::unique_ptr<NodeDomain>(new NodeDomain(domain));
  assert(node_domain);
  node->annot(std::move(node_domain));
}

References loco::AnnotatedItem< Annotation >::annot(), and loco::Unknown.

as_loco_index()

loco::TensorIndex locomotiv::as_loco_index

(

const nncc::core::ADT::tensor::Index &

index

)

Definition at line 34 of file ConvertIndex.cpp.

{
  loco::TensorIndex loco_index;
  loco_index.resize(index.rank());
  for (uint32_t axis = 0; axis < index.rank(); ++axis)
    loco_index.at(axis) = index.at(axis);
 
  return loco_index;
}

References loco::TensorIndex::at(), and loco::TensorIndex::resize().

as_nncc_index()

nncc::core::ADT::tensor::Index locomotiv::as_nncc_index

(

const loco::TensorIndex &

index

)

Definition at line 24 of file ConvertIndex.cpp.

{
  nncc::core::ADT::tensor::Index nncc_index;
  nncc_index.resize(index.rank());
  for (uint32_t axis = 0; axis < index.rank(); ++axis)
    nncc_index.at(axis) = index.at(axis);
 
  return nncc_index;
}

References nncc::core::ADT::tensor::Index::at(), and nncc::core::ADT::tensor::Index::resize().

erase_annot_domain()

void locomotiv::erase_annot_domain

(

loco::Node *

node

)

Erase already annotated node domain.

Definition at line 51 of file NodeDomain.cpp.

{ node->annot<NodeDomain>(nullptr); }

References loco::AnnotatedItem< Annotation >::annot().

Referenced by locomotiv::Session::~Session().

make_data() [1/3]

template<typename DT >

std::unique_ptr< NodeData > locomotiv::make_data

(

const NodeData::Buffer< DT > &

buffer

)

Copy buffer to make NodeData.

Note

NodeData is read-only. You may prepare buffer with ALL data, then call this function to make data.

References make_data().

Referenced by make_data(), make_data(), make_data(), and nnkit::support::moco::tf::Backend::prepare().

make_data() [2/3]

template<>

std::unique_ptr< NodeData > locomotiv::make_data

(

const NodeData::Buffer< float > &

buf

)

Definition at line 28 of file NodeData.cpp.

{
  return std::unique_ptr<NodeDataImpl>(new NodeDataImpl(buf));
}

References make_data().

make_data() [3/3]

template<>

std::unique_ptr< NodeData > locomotiv::make_data

(

const NodeData::Buffer< int32_t > &

buf

)

Definition at line 23 of file NodeData.cpp.

{
  return std::unique_ptr<NodeDataImpl>(new NodeDataImpl(buf));
}

References make_data().

validate()

void locomotiv::validate ( bool  true_cond, const std::string &&  exception_msg  )

inline

Definition at line 26 of file Validation.h.

{
  if (!true_cond)
    throw std::runtime_error(exception_msg);
}