Descriptor for an inner product weights gradient primitive. More...
#include <dnnl.hpp>

Public Member Functions | |
| desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc) | |
| Constructs a descriptor for an inner product descriptor weights update primitive with bias. More... | |
| desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc) | |
| Constructs a descriptor for an inner product descriptor weights update primitive without bias. More... | |
Descriptor for an inner product weights gradient primitive.
|
inline |
Constructs a descriptor for an inner product descriptor weights update primitive with bias.
Inputs:
src (dnnl::primitive_desc_base::src_desc(0))diff_dst (dnnl::primitive_desc_base::diff_dst_desc(0))Outputs:
diff_weights (dnnl::primitive_desc_base::diff_weights_desc(0))diff_bias (dnnl::primitive_desc_base::diff_weights_desc(1))format_tag.| src_desc | Memory descriptor for src. |
| diff_weights_desc | Memory descriptor for diff weights. |
| diff_bias_desc | Memory descriptor for diff bias. |
| diff_dst_desc | Memory descriptor for diff dst. |
|
inline |
Constructs a descriptor for an inner product descriptor weights update primitive without bias.
Inputs:
src (dnnl::primitive_desc_base::src_desc(0))diff_dst (dnnl::primitive_desc_base::diff_dst_desc(0))Outputs:
diff_weights (dnnl::primitive_desc_base::diff_weights_desc(0))format_tag.| src_desc | Memory descriptor for src. |
| diff_weights_desc | Memory descriptor for diff weights. |
| diff_dst_desc | Memory descriptor for diff dst. |