Shuffle

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

Functions
mkldnn_status_t MKLDNN_API	mkldnn_shuffle_forward_desc_init (mkldnn_shuffle_desc_t *shuffle_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, int axis, mkldnn_dim_t group_size)
	Initializes a shuffle_desc for forward propagation using prop_kind, memory descriptor data_desc, axis, and group_size. More...
mkldnn_status_t MKLDNN_API	mkldnn_shuffle_backward_desc_init (mkldnn_shuffle_desc_t *shuffle_desc, const mkldnn_memory_desc_t *diff_data_desc, int axis, mkldnn_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.

See also

Shuffle in developer guide

Shuffle in C++ API

Function Documentation

mkldnn_shuffle_forward_desc_init()

mkldnn_status_t MKLDNN_API mkldnn_shuffle_forward_desc_init	(	mkldnn_shuffle_desc_t *	shuffle_desc,
		mkldnn_prop_kind_t	prop_kind,
		const mkldnn_memory_desc_t *	data_desc,
		int	axis,
		mkldnn_dim_t	group_size
	)

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

Inputs:

• src (mkldnn_query_src_md, 0)

Outputs:

• dst (mkldnn_query_dst_md, 0)

mkldnn_shuffle_backward_desc_init()

mkldnn_status_t MKLDNN_API mkldnn_shuffle_backward_desc_init	(	mkldnn_shuffle_desc_t *	shuffle_desc,
		const mkldnn_memory_desc_t *	diff_data_desc,
		int	axis,
		mkldnn_dim_t	group_size
	)

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

Inputs:

• diff_dst (mkldnn_query_diff_dst_md, 0)

Outputs:

• diff_src (mkldnn_query_diff_src_md, 0)