mxnet
|
A primitive to compute deconvolution using different algorithms.
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_backward_data_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | diff_src_desc, | ||
const mkldnn_memory_desc_t * | weights_desc, | ||
const mkldnn_memory_desc_t * | diff_dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a deconvolution descriptor conv_desc
for backward propagation with respect to data using alg_kind
, memory descriptors, strides
, padding_l
, padding_r
, and padding_kind
.
format_kind
.Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_backward_weights_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | src_desc, | ||
const mkldnn_memory_desc_t * | diff_weights_desc, | ||
const mkldnn_memory_desc_t * | diff_bias_desc, | ||
const mkldnn_memory_desc_t * | diff_dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a deconvolution descriptor conv_desc
for backward propagation with respect to weights using alg_kind
, memory descriptors, strides
, padding_l
, padding_r
, and padding_kind
.
format_kind
.Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_forward_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_prop_kind_t | prop_kind, | ||
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | src_desc, | ||
const mkldnn_memory_desc_t * | weights_desc, | ||
const mkldnn_memory_desc_t * | bias_desc, | ||
const mkldnn_memory_desc_t * | dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a deconvolution descriptor deconv_desc
for forward propagation using prop_kind
(possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind
, memory descriptors, strides
, padding_l
, padding_r
, and padding_kind
. In order to create a deconvolution without bias, bias_desc
should either be NULL
or point to a descriptor with memory format equal to mkldnn_format_undef.
padding_r
is NULL
, the padding is supposed to be symmetric.format_kind
.Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_backward_data_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | diff_src_desc, | ||
const mkldnn_memory_desc_t * | weights_desc, | ||
const mkldnn_memory_desc_t * | diff_dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | dilates, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a dilated deconvolution descriptor conv_desc
for backward propagation with respect to data using alg_kind
, memory descriptors, strides
, dilates
, padding_l
, padding_r
, and padding_kind
.
format_kind
.Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_backward_weights_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | src_desc, | ||
const mkldnn_memory_desc_t * | diff_weights_desc, | ||
const mkldnn_memory_desc_t * | diff_bias_desc, | ||
const mkldnn_memory_desc_t * | diff_dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | dilates, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a dilated deconvolution descriptor conv_desc
for backward propagation with respect to weights using alg_kind
, memory descriptors, strides
, dilates
, padding_l
, padding_r
, and padding_kind
.
format_kind
.Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_forward_desc_init | ( | mkldnn_deconvolution_desc_t * | conv_desc, |
mkldnn_prop_kind_t | prop_kind, | ||
mkldnn_alg_kind_t | alg_kind, | ||
const mkldnn_memory_desc_t * | src_desc, | ||
const mkldnn_memory_desc_t * | weights_desc, | ||
const mkldnn_memory_desc_t * | bias_desc, | ||
const mkldnn_memory_desc_t * | dst_desc, | ||
const mkldnn_dims_t | strides, | ||
const mkldnn_dims_t | dilates, | ||
const mkldnn_dims_t | padding_l, | ||
const mkldnn_dims_t | padding_r, | ||
mkldnn_padding_kind_t | padding_kind | ||
) |
Initializes a dilated deconvolution descriptor deconv_desc
for forward propagation using prop_kind
(possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind
, memory descriptors, strides
, dilates
, padding_l
, padding_r
, and padding_kind
. In order to create a dilated deconvolution without bias, bias_desc
should either be NULL
or point to a descriptor with memory format equal to mkldnn_format_undef.
padding_r
is NULL
, the padding is supposed to be symmetric.format_kind
.Order of inputs:
Order of outputs: