dnnl::matmul::desc Struct Reference

Descriptor for a matmul primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::matmul::desc:

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.
	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.

Detailed Description

Descriptor for a matmul primitive.

Examples

cpu_matmul_quantization.cpp, cpu_sgemm_and_matmul.cpp, inference_int8_matmul.cpp, and matmul.cpp.

Constructor & Destructor Documentation

desc() [1/2]

dnnl::matmul::desc::desc ( const memory::desc & src_desc, const memory::desc & weights_desc, const memory::desc & dst_desc )

inline

Constructs a descriptor for a matmul primitive.

Inputs:

• src (dnnl::primitive_desc_base::src_desc(0))
• weights (dnnl::primitive_desc_base::weights_desc(0))

Outputs:

• dst (dnnl::primitive_desc_base::dst_desc(0))

Parameters

src_desc	Memory descriptor for source (matrix A).
weights_desc	Memory descriptor for weights (matrix B).
dst_desc	Memory descriptor for destination (matrix C).

desc() [2/2]

dnnl::matmul::desc::desc ( const memory::desc & src_desc, const memory::desc & weights_desc, const memory::desc & bias_desc, const memory::desc & dst_desc )

inline

Constructs a descriptor for a matmul primitive.

Inputs:

• src (dnnl::primitive_desc_base::src_desc(0))
• weights (dnnl::primitive_desc_base::weights_desc(0))
• bias (dnnl::primitive_desc_base::weights_desc(1))

Outputs:

• dst (dnnl::primitive_desc_base::dst_desc(0))

Parameters

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.

---

The documentation for this struct was generated from the following file:

• include/dnnl.hpp