mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-25 03:30:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Alexander Alekhin 2020-02-22 22:21:56 +00:00
commit 8eafddf211
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
View File

@ -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())