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.
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.
format_tag
.src_desc | Memory descriptor for src. |
diff_weights_desc | Memory descriptor for diff weights. |
diff_dst_desc | Memory descriptor for diff dst. |