A primitive to compute resampling operation on 1D, 2D or 3D data tensor using Nearest Neighbor, or Linear (Bilinear, Trilinear) interpolation method. More...
Classes | |
struct | dnnl::resampling_forward |
Resampling forward propagation. More... | |
struct | dnnl::resampling_backward |
Resampling backward propagation primitive. More... | |
struct | dnnl_resampling_desc_t |
A descriptor of resampling operation. More... | |
Functions | |
dnnl_status_t DNNL_API | dnnl_resampling_forward_desc_init (dnnl_resampling_desc_t *resampling_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const float *factors, const dnnl_memory_desc_t *src_desc, const dnnl_memory_desc_t *dst_desc) |
Initializes a descriptor for a resampling forward propagation primitive. More... | |
dnnl_status_t DNNL_API | dnnl_resampling_backward_desc_init (dnnl_resampling_desc_t *resampling_desc, dnnl_alg_kind_t alg_kind, const float *factors, const dnnl_memory_desc_t *diff_src_desc, const dnnl_memory_desc_t *diff_dst_desc) |
Initializes a descriptor for resampling backward propagation primitive. More... | |
A primitive to compute resampling operation on 1D, 2D or 3D data tensor using Nearest Neighbor, or Linear (Bilinear, Trilinear) interpolation method.
dnnl_status_t DNNL_API dnnl_resampling_forward_desc_init | ( | dnnl_resampling_desc_t * | resampling_desc, |
dnnl_prop_kind_t | prop_kind, | ||
dnnl_alg_kind_t | alg_kind, | ||
const float * | factors, | ||
const dnnl_memory_desc_t * | src_desc, | ||
const dnnl_memory_desc_t * | dst_desc | ||
) |
Initializes a descriptor for a resampling forward propagation primitive.
resampling_desc | Output descriptor for a resampling primitive. |
prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
alg_kind | resampling algorithm kind: either dnnl_resampling_nearest, or dnnl_resampling_linear. |
factors | Array of scaling factors for spatial dimension. |
src_desc | Source memory descriptor. |
dst_desc | Destination memory descriptor. |
dnnl_status_t DNNL_API dnnl_resampling_backward_desc_init | ( | dnnl_resampling_desc_t * | resampling_desc, |
dnnl_alg_kind_t | alg_kind, | ||
const float * | factors, | ||
const dnnl_memory_desc_t * | diff_src_desc, | ||
const dnnl_memory_desc_t * | diff_dst_desc | ||
) |
Initializes a descriptor for resampling backward propagation primitive.
resampling_desc | Output descriptor for a resampling primitive. |
alg_kind | resamplinging algorithm kind: either dnnl_resampling_nearest, or dnnl_resampling_linear. |
diff_src_desc | Diff source memory descriptor. |
diff_dst_desc | Diff destination memory descriptor. |
factors | Array of scaling factors for spatial dimension. |