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

Detailed Description

Descriptor for a matmul primitive.

Examples:

cpu_matmul_quantization.cpp, cpu_sgemm_and_matmul.cpp, and inference_int8_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