fix weights rank assertion in InnerProductOp

c23ab373 · YashasSamaga · 96b26dc8 · c23ab373
Commit c23ab373 authored Feb 22, 2020 by YashasSamaga
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

inner_product.hpp modules/dnn/src/cuda4dnn/primitives/inner_product.hpp +1 -1

No files found.
--- 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())