oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
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

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.

Parameters
src_descMemory descriptor for source (matrix A).
weights_descMemory descriptor for weights (matrix B).
dst_descMemory 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.

Parameters
src_descMemory descriptor for source (matrix A).
weights_descMemory descriptor for weights (matrix B).
dst_descMemory descriptor for destination (matrix C).
bias_descMemory descriptor for bias.

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