split config idx fix (#10)

mandrono · Jul 14, 2021 · 46e8c85 · 46e8c85
1 parent a15a55d
commit 46e8c85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_split_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_split_node.cpp
@@ -437,7 +437,7 @@ void MKLDNNSplitNode::selectOptimalPrimitiveDescriptor() {
             auto childEdge = getChildEdgeAt(i);
             auto childPtr = childEdge->getChild();
             auto& vecChildSpd = childPtr->getSupportedPrimitiveDescriptors();
-            const auto& outputDesc = supportedPrimitiveDescriptors[indx].getConfig().outConfs[i].desc;
+            const auto& outputDesc = supportedPrimitiveDescriptors[indx].getConfig().outConfs[childEdge->getInputNum()].desc;
 
             if (!vecChildSpd.empty()) {
                 int inNum = childEdge->getOutputNum();