Dnnl_api_concat

Collaboration diagram for Dnnl_api_concat:

Functions
dnnl_status_t DNNL_API	dnnl_concat_primitive_desc_create (dnnl_primitive_desc_t *concat_primitive_desc, const dnnl_memory_desc_t *dst_desc, int n, int concat_dimension, const dnnl_memory_desc_t *src_descs, const_dnnl_primitive_attr_t attr, dnnl_engine_t engine)

Detailed Description

Function Documentation

dnnl_status_t DNNL_API dnnl_concat_primitive_desc_create	(	dnnl_primitive_desc_t *	concat_primitive_desc,
		const dnnl_memory_desc_t *	dst_desc,
		int	n,
		int	concat_dimension,
		const dnnl_memory_desc_t *	src_descs,
		const_dnnl_primitive_attr_t	attr,
		dnnl_engine_t	engine
	)

Creates a primitive descriptor for an out-of-place concatenation primitive.

Parameters

concat_primitive_desc	Output primitive descriptor.
dst_desc	Destination memory descriptor.
n	Number of source parameters.
concat_dimension	Source tensors will be concatenated over dimension with this index. Note that order of dimensions does not depend on memory format.
src_descs	Array of source memory descriptors with n elements.
attr	Primitive attributes to use (can be NULL).
engine	Engine to use.

Returns

dnnl_success on success and a status describing the error otherwise.