Concat

Collaboration diagram for Concat:

Functions
mkldnn_status_t MKLDNN_API	mkldnn_concat_primitive_desc_create (mkldnn_primitive_desc_t *concat_primitive_desc, const mkldnn_memory_desc_t *dst_md, int n, int concat_dimension, const mkldnn_memory_desc_t *src_mds, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine)

Detailed Description

A primitive to concatenate data by arbitrary dimension.

See also

dev_guide_concat in developer guide

cpp_api_concat in cpp_api

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_concat_primitive_desc_create	(	mkldnn_primitive_desc_t *	concat_primitive_desc,
		const mkldnn_memory_desc_t *	dst_md,
		int	n,
		int	concat_dimension,
		const mkldnn_memory_desc_t *	src_mds,
		const_mkldnn_primitive_attr_t	attr,
		mkldnn_engine_t	engine
	)

Creates out-of-place concat_primitive_desc for concatenation of n inputs by concat_dimension with resulting output_desc memory descriptor. output_desc can be NULL or specified with the mkldnn_format_kind_any format kind – in this case, the appropriate memory format would be chosen automatically.

Inputs:

• input 0 (mkldnn_query_src_md, 0)
• input 1 (mkldnn_query_src_md, 1)
• ...
• input n - 1 (mkldnn_query_src_md, n - 1)

Outputs:

• output (mkldnn_query_dst_md, 0)