mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-07-25 22:57:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix weights rank assertion in InnerProductOp

Browse Source
This commit is contained in:
YashasSamaga 2020-02-22 16:59:09 +05:30
parent 96b26dc8f4
commit c23ab37355
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())