Merge pull request #16647 from YashasSamaga:cuda4dnn-hotfix-i16568

5 years ago · 8eafddf211
parent 102ef39a27 c23ab37355
commit 8eafddf211
1 changed files with 1 additions and 1 deletions
--- a/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
+++ b/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
@ -31,7 +31,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
            : stream(std::move(stream_)), cublasHandle(std::move(handle)), axis{ axis }
        {
            weightsTensor = csl::makeTensorHeader<T>(weights);
-            CV_Assert(get_effective_rank(weightsTensor) == 2);
+            CV_Assert(get_effective_rank(weightsTensor) <= 2);
            csl::copyMatToTensor<T>(weights, weightsTensor, stream);

            if (!bias.empty())