Skip to content

N
ngraph
Commit 2f0a262e authored by Matthew Brookhart's avatar Matthew Brookhart Committed by Scott Cyphers
Browse files
Options

Fix a logic bug introduced by #325 (#347)

parent d901282e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/ngraph/util.cpp
View file @ 2f0a262e
......@@ -183,8 +183,8 @@ ngraph::FpropCache ngraph::cache_fprop(std::shared_ptr<ngraph::Function> fprop,
// shape and element type as the nodes in fprop
NodeMap node_param_map;
ngraph::traverse_nodes(fprop, [&node_param_map](std::shared_ptr<Node> node) {
node_param_map.get(
std::make_shared<op::Parameter>(node->get_element_type(), node->get_shape()));
node_param_map.add(
node, std::make_shared<op::Parameter>(node->get_element_type(), node->get_shape()));
});
// Traverse bprop to find all of the nodes in the graph
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment