Descriptor for a convolution weights gradient primitive. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (algorithm algorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &padding_l, const memory::dims &padding_r) | |
Constructs a descriptor for a convolution weights gradient primitive with bias. More... | |
desc (algorithm algorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &padding_l, const memory::dims &padding_r) | |
Constructs a descriptor for a convolution weights gradient primitive without bias. More... | |
desc (algorithm algorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &dilates, const memory::dims &padding_l, const memory::dims &padding_r) | |
Constructs a descriptor for a dilated convolution weights gradient primitive with bias. More... | |
desc (algorithm algorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &dilates, const memory::dims &padding_l, const memory::dims &padding_r) | |
Constructs a descriptor for a dilated convolution weights gradient primitive without bias. More... | |
Descriptor for a convolution weights gradient primitive.
|
inline |
Constructs a descriptor for a convolution weights gradient primitive with bias.
format_tag
.Inputs:
Outputs:
algorithm | Convolution algorithm. Possible values are dnnl::algorithm::convolution_direct, dnnl::algorithm::convolution_winograd, and dnnl::algorithm::convolution_auto. |
src_desc | Source memory descriptor. |
diff_weights_desc | Diff weights memory descriptor. |
diff_bias_desc | Diff bias memory descriptor. Passing zero memory descriptor disables the bias term. |
diff_dst_desc | Diff destination memory descriptor. |
strides | Strides for each spatial dimension. |
padding_l | Vector of padding values for low indices for each spatial dimension (front, top, left). |
padding_r | Vector of padding values for high indices for each spatial dimension (back, bottom, right). |
|
inline |
Constructs a descriptor for a convolution weights gradient primitive without bias.
format_tag
.Inputs:
Outputs:
algorithm | Convolution algorithm. Possible values are dnnl::algorithm::convolution_direct, dnnl::algorithm::convolution_winograd, and dnnl::algorithm::convolution_auto. |
src_desc | Source memory descriptor. |
diff_weights_desc | Diff weights memory descriptor. |
diff_dst_desc | Diff destination memory descriptor. |
strides | Strides for each spatial dimension. |
padding_l | Vector of padding values for low indices for each spatial dimension (front, top, left). |
padding_r | Vector of padding values for high indices for each spatial dimension (back, bottom, right). |
|
inline |
Constructs a descriptor for a dilated convolution weights gradient primitive with bias.
format_tag
.Inputs:
Outputs:
algorithm | Convolution algorithm. Possible values are dnnl::algorithm::convolution_direct, dnnl::algorithm::convolution_winograd, and dnnl::algorithm::convolution_auto. |
src_desc | Source memory descriptor. |
diff_weights_desc | Diff weights memory descriptor. |
diff_bias_desc | Diff bias memory descriptor. Passing zero memory descriptor disables the bias term. |
diff_dst_desc | Diff destination memory descriptor. |
strides | Strides for each spatial dimension. |
dilates | Dilations for each spatial dimension. A zero value means no dilation in the corresponding dimension. |
padding_l | Vector of padding values for low indices for each spatial dimension (front, top, left). |
padding_r | Vector of padding values for high indices for each spatial dimension (back, bottom, right). |
|
inline |
Constructs a descriptor for a dilated convolution weights gradient primitive without bias.
format_tag
.Inputs:
Outputs:
algorithm | Convolution algorithm. Possible values are dnnl::algorithm::convolution_direct, dnnl::algorithm::convolution_winograd, and dnnl::algorithm::convolution_auto. |
src_desc | Source memory descriptor. |
diff_weights_desc | Diff weights memory descriptor. |
diff_dst_desc | Diff destination memory descriptor. |
strides | Strides for each spatial dimension. |
dilates | Dilations for each spatial dimension. A zero value means no dilation in the corresponding dimension. |
padding_l | Vector of padding values for low indices for each spatial dimension (front, top, left). |
padding_r | Vector of padding values for high indices for each spatial dimension (back, bottom, right). |