added index that check all inputs of nodes that

match
This commit is contained in:
Abduragim Shtanchaev 2023-05-29 14:48:42 +03:00
parent 02397ef851
commit ecd2e8ff47

View File

@ -808,7 +808,7 @@ public:
const Ptr<ImportNodeWrapper> node_to_check = net->getNode(i); const Ptr<ImportNodeWrapper> node_to_check = net->getNode(i);
int numInp = node_to_check->getNumInputs(); int numInp = node_to_check->getNumInputs();
for (int inp = 0; inp < numInp; ++inp) { for (int inp = 0; inp < numInp; ++inp) {
if (i != nodeToMatch && inpNodeName == node_to_check->getInputName(0)) { if (i != nodeToMatch && inpNodeName == node_to_check->getInputName(inp)) {
// Another node has the same input node, so it cannot be merged. // Another node has the same input node, so it cannot be merged.
return false; return false;
} }