Deep Neural Network Library (DNNL)  1.3.0
Performance library for Deep Learning
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 prop_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 prop_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.

Constructor & Destructor Documentation

◆ desc() [1/2]

dnnl::inner_product_forward::desc::desc ( prop_kind  prop_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
Memory descriptors can be initialized with dnnl::memory::format_tag::any value of format_tag.

Inputs:

Outputs:

Parameters
prop_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  prop_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
Memory descriptors can be initialized with dnnl::memory::format_tag::any value of format_tag.

Inputs:

Outputs:

Parameters
prop_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: