struct dnnl::matmul::desc
Overview
Descriptor for a matmul primitive. More…
#include <dnnl.hpp> struct desc { // fields dnnl_matmul_desc_t data; // construction desc( const memory::desc& src_desc, const memory::desc& weights_desc, const memory::desc& dst_desc ); desc( const memory::desc& src_desc, const memory::desc& weights_desc, const memory::desc& bias_desc, const memory::desc& dst_desc ); };
Detailed Documentation
Descriptor for a matmul primitive.
Construction
desc( const memory::desc& src_desc, const memory::desc& weights_desc, const memory::desc& dst_desc )
Constructs a descriptor for a matmul primitive.
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( 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.
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. |