Descriptor for a pooling backward propagation primitive. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (algorithm aalgorithm, const memory::desc &diff_src_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &kernel, const memory::dims &dilation, const memory::dims &padding_l, const memory::dims &padding_r) | |
Constructs a descriptor for pooling v2 (dilated pooling) backward propagation primitive. More... | |
Descriptor for a pooling backward propagation primitive.
|
inline |
Constructs a descriptor for pooling v2 (dilated pooling) backward propagation primitive.
Arrays strides
, kernel
, dilation
, padding_l
and padding_r
contain values for spatial dimensions only and hence must have the same number of elements as there are spatial dimensions. The order of values is the same as in the tensor: depth (for 3D tensors), height (for 3D and 2D tensors), and width.
aalgorithm | Pooling algorithm kind: either dnnl::algorithm::pooling_max, dnnl::algorithm::pooling_avg_include_padding, or dnnl::algorithm::pooling_avg (same as dnnl::algorithm::pooling_avg_exclude_padding). |
diff_src_desc | Diff source memory descriptor. |
diff_dst_desc | Diff destination memory descriptor. |
strides | Vector of strides for spatial dimension. |
kernel | Vector of kernel spatial dimensions. |
dilation | Array of dilations for 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) . |