1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 15:53:03 +04:00

Merge pull request #23061 from WanliZhong:gemm_cuda

DNN: make GEMM can be supported with transA and transB in CUDA
This commit is contained in:
Alexander Alekhin
2023-02-09 00:06:32 +03:00
committed by GitHub
5 changed files with 69 additions and 27 deletions
@@ -374,6 +374,10 @@ CV__DNN_INLINE_NS_BEGIN
static Ptr<SoftmaxLayerInt8> create(const LayerParams& params);
};
/**
* `InnerProduct`, `MatMul` and `Gemm` operations are all implemented by Fully Connected Layer.
* Parameter `is_matmul` is used to distinguish `MatMul` and `Gemm` from `InnerProduct`.
*/
class CV_EXPORTS InnerProductLayer : public Layer
{
public: