A primitive to shuffle data along the axis. More...

Functions
dnnl_status_t DNNL_API	dnnl_shuffle_forward_desc_init (dnnl_shuffle_desc_t shuffle_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, int axis, dnnl_dim_t group_size)
	Initializes a `shuffle_desc` for forward propagation using `prop_kind`, memory descriptor `data_desc`, `axis`, and `group_size`. More...

dnnl_status_t DNNL_API	dnnl_shuffle_backward_desc_init (dnnl_shuffle_desc_t shuffle_desc, const dnnl_memory_desc_t diff_data_desc, int axis, dnnl_dim_t group_size)
	Initializes a `shuffle_desc` for backward propagation using memory descriptor `diff_data_desc`, `axis`, and `group_size`. More...

Detailed Description

A primitive to shuffle data along the axis.

Function Documentation

dnnl_status_t DNNL_API dnnl_shuffle_forward_desc_init	(	dnnl_shuffle_desc_t *	shuffle_desc,
		dnnl_prop_kind_t	prop_kind,
		const dnnl_memory_desc_t *	data_desc,
		int	axis,
		dnnl_dim_t	group_size
	)

Initializes a shuffle_desc for forward propagation using prop_kind, memory descriptor data_desc, axis, and group_size.

Inputs:

Outputs:

dnnl_status_t DNNL_API dnnl_shuffle_backward_desc_init	(	dnnl_shuffle_desc_t *	shuffle_desc,
		const dnnl_memory_desc_t *	diff_data_desc,
		int	axis,
		dnnl_dim_t	group_size
	)

Initializes a shuffle_desc for backward propagation using memory descriptor diff_data_desc, axis, and group_size.

Inputs:

Outputs: