Collaboration diagram for Concat:


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.