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. |