Descriptor for a matmul primitive. More...
#include <dnnl.hpp>

Public Member Functions | |
| desc (const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc) | |
| Constructs a descriptor for a matmul primitive. More... | |
| desc (const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc) | |
| Constructs a descriptor for a matmul primitive. More... | |
Descriptor for a matmul primitive.
|
inline |
Constructs a descriptor for a matmul primitive.
| src_desc | Memory descriptor for source (matrix A). |
| weights_desc | Memory descriptor for weights (matrix B). |
| dst_desc | Memory descriptor for destination (matrix C). |
|
inline |
Constructs a descriptor for a matmul primitive.
| src_desc | Memory descriptor for source (matrix A). |
| weights_desc | Memory descriptor for weights (matrix B). |
| dst_desc | Memory descriptor for destination (matrix C). |
| bias_desc | Memory descriptor for bias. |