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

8eafddf2 · Alexander Alekhin · 102ef39a · c23ab373 · 8eafddf2
Commit 8eafddf2 authored Feb 22, 2020 by Alexander Alekhin
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())