IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Improve IsNodeOnConstPath() method (#7458)

Browse Source
This commit is contained in:
mei, yang 2021-09-22 15:52:58 +08:00 committed by GitHub
parent 377f46898c
commit e2272331be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
inference-engine/src/cldnn_engine/cldnn_program.cpp
View File

@ -325,28 +325,24 @@ void Program::InitProfileInfo(const std::string& layerName,
// TODO: Does it make sense to add such method to ngraph core?
bool IsNodeOnConstPath(const std::shared_ptr<ngraph::Node>& node) {
std::list<std::shared_ptr<ngraph::Node>> nodes_to_process = { node };
while (!nodes_to_process.empty()) {
auto current_node = nodes_to_process.front();
nodes_to_process.pop_front();
for (size_t i = 0; i < current_node->get_input_size(); i++) {
auto input_node = current_node->get_input_node_shared_ptr(i);
std::set<std::shared_ptr<ngraph::Node>> nodes_processed = {};
std::function<bool(const std::shared_ptr<ngraph::Node>&)> is_const_node = [&nodes_processed, &is_const_node](const std::shared_ptr<ngraph::Node>& node) {
if (nodes_processed.count(node)) return true;
nodes_processed.insert(node);
// If input is constant, then drop if from the processing list
if (std::dynamic_pointer_cast<ngraph::op::v0::Constant>(input_node) != nullptr)
continue;
if (std::dynamic_pointer_cast<ngraph::op::v0::Constant>(node) != nullptr)
return true;
// If the node doesn't have any parents and it's not a constant, then we deal with dynamic path
if (input_node->get_input_size() == 0) {
if (node->get_input_size() == 0)
return false;
for (size_t i = 0; i < node->get_input_size(); i++) {
auto input_node = node->get_input_node_shared_ptr(i);
if (!is_const_node(input_node))
return false;
}
nodes_to_process.insert(nodes_to_process.end(), input_node);
}
}
return true;
};
return is_const_node(node);
}
} // namespace CLDNNPlugin