moco::NodeDefTable Class Reference

Class to store and query tensorflow::NodeDef* with string name key. More...

#include <GraphBuilderContext.h>

Public Member Functions
void	enroll (const std::string &node_name, const tensorflow::NodeDef *node)
	Registers a name with corresponding tensorflow::NodeDef*.
const tensorflow::NodeDef *	node (const std::string &node_name) const
	Queries enrolled(registered) with name and return node if found Will throw runtime_error if not found.

Detailed Description

Class to store and query tensorflow::NodeDef* with string name key.

Definition at line 37 of file GraphBuilderContext.h.

Member Function Documentation

enroll()

void moco::NodeDefTable::enroll	(	const std::string &	node_name,
		const tensorflow::NodeDef *	node
	)

Registers a name with corresponding tensorflow::NodeDef*.

Definition at line 27 of file GraphBuilderContext.cpp.

{
  MapNameNode_t::iterator iter = _table.find(node_name);
 
  if (iter != _table.end())
  {
    throw oops::UserExn("Duplicate node name in GraphDef", node_name);
  }
 
  _table[node_name] = node;
}

References node().

node()

const tensorflow::NodeDef * moco::NodeDefTable::node

(

const std::string &

node_name

)

const

Queries enrolled(registered) with name and return node if found Will throw runtime_error if not found.

Definition at line 39 of file GraphBuilderContext.cpp.

{
  MapNameNode_t::const_iterator iter = _table.find(node_name);
 
  if (iter == _table.end())
  {
    throw oops::UserExn("Cannot find node with name in GraphDef", node_name);
  }
 
  return iter->second;
}

Referenced by enroll().

---

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

• compiler/moco/import/include/moco/Import/GraphBuilderContext.h
• compiler/moco/import/src/GraphBuilderContext.cpp