Skip to content

O
opencv
Commit 8eafddf2 authored by Alexander Alekhin's avatar Alexander Alekhin
Browse files
Options

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

parents 102ef39a c23ab373
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
View file @ 8eafddf2
...@@ -31,7 +31,7 @@ namespace cv { namespace dnn { namespace cuda4dnn { ...@@ -31,7 +31,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
: stream(std::move(stream_)), cublasHandle(std::move(handle)), axis{ axis } : stream(std::move(stream_)), cublasHandle(std::move(handle)), axis{ axis }
{ {
weightsTensor = csl::makeTensorHeader<T>(weights); 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); csl::copyMatToTensor<T>(weights, weightsTensor, stream);
if (!bias.empty()) if (!bias.empty())
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment