Descriptor for an inner product forward propagation primitive. More...
#include <dnnl.hpp>
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... | |
Descriptor for an inner product forward propagation primitive.
|
inline |
Constructs a descriptor for an inner product forward propagation primitive with bias.
format_tag
.Inputs:
Outputs:
prop_kind | Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference. |
src_desc | Memory descriptor for src. |
weights_desc | Memory descriptor for diff weights. |
bias_desc | Memory descriptor for diff bias. |
dst_desc | Memory descriptor for diff dst. |
|
inline |
Constructs a descriptor for an inner product forward propagation primitive without bias.
format_tag
.Inputs:
Outputs:
prop_kind | Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference. |
src_desc | Memory descriptor for src. |
weights_desc | Memory descriptor for diff weights. |
dst_desc | Memory descriptor for dst. |