potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 15:53:03 +04:00
Code Packages Releases Wiki Activity

Gather-Cast, Mul-Cast fusion

Browse Source
This commit is contained in:
ashishiva3@gmail.com
2020-03-01 15:09:15 +05:30
parent 4d0f13544d
commit e18d5e94c7
3 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/graph_simplifier.cpp
+2 -3
View File
@@ -194,15 +194,14 @@ void simplifySubgraphs(const Ptr<ImportGraphWrapper>& net,
{
int numNodes = net->getNumNodes();
std::vector<int> matchedNodesIds, targetNodesIds;
for (int i = 0; i < numNodes; ++i)
for (int j = 0; j < patterns.size(); ++j)
{
for (int j = 0; j < patterns.size(); ++j)
for (int i = 0; i < numNodes; ++i)
{
if (patterns[j]->match(net, i, matchedNodesIds, targetNodesIds))
{
patterns[j]->replace(net, matchedNodesIds, targetNodesIds);
numNodes -= matchedNodesIds.size() - 1; // #matchedNodes removed and one added.
break;
}
}
}