29 MapNameNode_t::iterator iter = _table.find(node_name);
31 if (iter != _table.end())
33 throw oops::UserExn(
"Duplicate node name in GraphDef", node_name);
36 _table[node_name] =
node;
41 MapNameNode_t::const_iterator iter = _table.find(node_name);
43 if (iter == _table.end())
45 throw oops::UserExn(
"Cannot find node with name in GraphDef", node_name);
53 MapNameNode_t::iterator iter = _table.find(tensor_name);
55 if (iter != _table.end())
57 throw oops::UserExn(
"Duplicate node name in GraphDef", tensor_name.name());
60 _table[tensor_name] =
node;
65 MapNameNode_t::const_iterator iter = _table.find(tensor_name);
67 if (iter == _table.end())
69 throw oops::UserExn(
"Cannot find node with name in GraphDef", tensor_name.name());
77 _queue.push_back(std::move(
update));
Logical unit of computation.
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...
void enroll(const std::string &node_name, const tensorflow::NodeDef *node)
Registers a name with corresponding tensorflow::NodeDef*.
void enroll(const TensorName &tensor_name, loco::Node *node)
Registers a name with corresponding loco::Node *.
loco::Node * node(const TensorName &tensor_name) const
Queries enrolled(registered) with name and return node if found Will throw runtime_error if not found...
void enroll(std::unique_ptr< GraphUpdate > &&update)
Registers GraphUpdate objects.
FeatureShapeUpdater update(loco::FeatureShape &feature_shape)