struct dnnl_convolution_desc_t¶
Overview¶
A descriptor of a convolution operation. More…
#include <dnnl_types.h> struct dnnl_convolution_desc_t { // fields dnnl_primitive_kind_t primitive_kind; dnnl_prop_kind_t prop_kind; dnnl_alg_kind_t alg_kind; dnnl_memory_desc_t src_desc; dnnl_memory_desc_t diff_src_desc; dnnl_memory_desc_t weights_desc; dnnl_memory_desc_t diff_weights_desc; dnnl_memory_desc_t bias_desc; dnnl_memory_desc_t diff_bias_desc; dnnl_memory_desc_t dst_desc; dnnl_memory_desc_t diff_dst_desc; dnnl_dims_t strides; dnnl_dims_t dilates; dnnl_dims_t padding[2]; dnnl_data_type_t accum_data_type; };
Detailed Documentation¶
A descriptor of a convolution operation.
Fields¶
dnnl_primitive_kind_t primitive_kind
The kind of primitive.
Used for self-identifying the primitive descriptor. Must be dnnl_convolution.
dnnl_prop_kind_t prop_kind
The kind of propagation.
Possible values: dnnl_forward_training, dnnl_forward_inference, dnnl_backward_data, dnnl_backward_weights, and dnnl_backward_bias.
dnnl_alg_kind_t alg_kind
The kind of the convolution algorithm.
Possible values: dnnl_convolution_direct.
dnnl_memory_desc_t src_desc
Source memory descriptor.
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
dnnl_memory_desc_t weights_desc
Weights memory descriptor.
dnnl_memory_desc_t diff_weights_desc
Weights gradient memory descriptor.
dnnl_memory_desc_t bias_desc
Bias memory descriptor.
dnnl_memory_desc_t diff_bias_desc
Bias gradient memory descriptor.
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
dnnl_dims_t strides
Convolution strides in each spatial dimension.
dnnl_dims_t dilates
Convolution dilates in each spatial dimension.
dnnl_dims_t padding[2]
Padding in each spatial dimension.
padding[0] is a padding in the beginning (padding_l), padding[1] is a padding in the end (padding_r).
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.