mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +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 } : 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())