From c23ab3735580877d13d1df60a2065277d6b7efd8 Mon Sep 17 00:00:00 2001
From: YashasSamaga <yashas_2010@yahoo.com>
Date: Sat, 22 Feb 2020 16:59:09 +0530
Subject: [PATCH] fix weights rank assertion in InnerProductOp

---
 modules/dnn/src/cuda4dnn/primitives/inner_product.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp b/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
index d5baa50b34f9..a0a00ce125c0 100644
--- 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())