struct dnnl::inner_product_forward::desc¶
Overview¶
Descriptor for an inner product forward propagation primitive. More…
#include <dnnl.hpp> struct desc { // fields dnnl_inner_product_desc_t data; // construction 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 ); desc( prop_kind aprop_kind, const memory::desc& src_desc, const memory::desc& weights_desc, const memory::desc& dst_desc ); };
Detailed Documentation¶
Descriptor for an inner product forward propagation primitive.
Construction¶
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.
Note
All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag
.
Parameters:
aprop_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. |
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.
Note
All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag
.
Parameters:
aprop_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. |