dnn: fix build with Vulkan

5 years ago · 65552bf403
parent 6fdce865b8
commit 65552bf403
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/layers/convolution_layer.cpp
+++ b/modules/dnn/src/layers/convolution_layer.cpp
@ -460,11 +460,11 @@ public:

        std::vector<Ptr<BackendWrapper> > blobsWrapper;

-        if (newWeightAndBias)
+        if (fusedWeights)
        {
            Mat wm;
            weightsMat.copyTo(wm); // to handle the case of isContinuous() == false
-            wm.reshape(1, blobs[0].dims, blobs[0].size);
+            wm = wm.reshape(1, blobs[0].dims, blobs[0].size);
            blobsWrapper.push_back(Ptr<BackendWrapper>(new VkComBackendWrapper(wm)));
        }
        else