mirror of
https://github.com/opencv/opencv.git
synced 2026-07-31 00:03:03 +04:00
make GEMM can be supported with transA and transB in CUDA
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user