Commit 7d058da2 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball

switching to using get_users() (#744)

parent 2bf2214f
......@@ -183,8 +183,9 @@ std::list<std::shared_ptr<ngraph::Node>>
result_list.push_back(node_map[independent_node]);
independent_nodes.pop_front();
for (auto user : independent_node->users())
for (auto user_sp : independent_node->get_users())
{
Node* user = user_sp.get();
node_dependency_count[user] -= 1;
size_t count = node_dependency_count[user];
if (count == 0)
......
......@@ -35,8 +35,6 @@ namespace ngraph
: Pattern("Any", NodeVector{arg}, predicate)
{
add_output(arg->get_element_type(), arg->get_shape());
//m_arguments.push_back(arg);
//const_cast<std::multiset<Node*>&>(arg->users()).insert(this);
}
};
}
......
......@@ -91,8 +91,9 @@ static vector<unordered_set<shared_ptr<Node>>>
previous_placement = independent_node->get_placement();
sorted_nodes.push_back(node_map.at(independent_node));
for (Node* user_node : independent_node->users())
for (auto user : independent_node->get_users())
{
Node* user_node = user.get();
node_dependency_count.at(user_node) -= 1;
if (node_dependency_count.at(user_node) == 0)
{
......
......@@ -304,8 +304,9 @@ static json write(const Function& f, bool binary_constant_data)
result_list.push_back(node_map[independent_node]);
independent_nodes.pop_front();
for (auto user : independent_node->users())
for (auto sp_user : independent_node->get_users())
{
Node* user = sp_user.get();
node_depencency_count[user] -= 1;
size_t count = node_depencency_count[user];
if (count == 0)
......
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