A primitive to perform matrix-matrix multiplication. The batched mode is supported with 3D tensors.
- See also
- Matrix Multiplication in developer guide
◆ dnnl_matmul_desc_init()
Initializes a matrix multiplication descriptor.
- Parameters
-
matmul_desc | Output descriptor for matmul primitive. |
src_desc | Source memory descriptor (matrix A) |
weights_desc | Weights memory descriptor (matrix B) |
bias_desc | Bias memory descriptor. Passing NULL, a zero memory descriptor, or a memory descriptor with format_kind set to dnnl_format_kind_undef disables the bias term. |
dst_desc | Destination memory descriptor (matrix C). |
- Returns
- dnnl_success on success and a status describing the error otherwise.