oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
Public Member Functions | List of all members
dnnl::inner_product_forward::desc Struct Reference

Descriptor for an inner product forward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::inner_product_forward::desc:
Collaboration graph
[legend]

Public Member Functions

 desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc)
 Constructs a descriptor for an inner product forward propagation primitive with bias. More...
 
 desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc)
 Constructs a descriptor for an inner product forward propagation primitive without bias. More...
 

Detailed Description

Descriptor for an inner product forward propagation primitive.

Examples:
cnn_inference_f32.cpp, and inner_product.cpp.

Constructor & Destructor Documentation

◆ desc() [1/2]

dnnl::inner_product_forward::desc::desc ( prop_kind  aprop_kind,
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 an inner product forward propagation primitive with bias.

Note
All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag.
Parameters
aprop_kindPropagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.
src_descMemory descriptor for src.
weights_descMemory descriptor for diff weights.
bias_descMemory descriptor for diff bias.
dst_descMemory descriptor for diff dst.

◆ desc() [2/2]

dnnl::inner_product_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc &  src_desc,
const memory::desc &  weights_desc,
const memory::desc &  dst_desc 
)
inline

Constructs a descriptor for an inner product forward propagation primitive without bias.

Note
All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag.
Parameters
aprop_kindPropagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.
src_descMemory descriptor for src.
weights_descMemory descriptor for diff weights.
dst_descMemory descriptor for dst.
The documentation for this struct was generated from the following file: