Commit 2a19db0f authored by Alexander Alekhin's avatar Alexander Alekhin

Merge pull request #16106 from dkurt:dnn_ie_ngraph_weights_fusion

parents 45f69313 c2ca3ee2
...@@ -553,10 +553,10 @@ public: ...@@ -553,10 +553,10 @@ public:
} }
else else
{ {
Mat newWeights = blobs[0].reshape(1, outCn); Mat newWeights;
Mat cvWeights = weightsMat.colRange(0, newWeights.cols); Mat cvWeights = weightsMat.colRange(0, blobs[0].total() / outCn);
cvWeights.copyTo(newWeights); cvWeights.copyTo(newWeights);
ieWeights = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, kernel_shape, blobs[0].data); ieWeights = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, kernel_shape, newWeights.data);
} }
} }
...@@ -2033,9 +2033,9 @@ public: ...@@ -2033,9 +2033,9 @@ public:
if (fusedWeights) if (fusedWeights)
{ {
int inpCn = blobs[0].size[0]; Mat newWeights;
Mat newWeights = blobs[0].reshape(1, inpCn);
transpose(weightsMat, newWeights); transpose(weightsMat, newWeights);
ieWeights = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, kernel_shape, newWeights.data);
} }
size_t batch = ieInpNode->get_shape()[0]; size_t batch = ieInpNode->get_shape()[0];
std::vector<size_t> out_shape = {batch, (size_t)numOutput}; std::vector<size_t> out_shape = {batch, (size_t)numOutput};
......
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