Concat

Overview

A primitive to concatenate data by arbitrary dimension. More…

// structs

struct dnnl::concat;

// global 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 Documentation

A primitive to concatenate data by arbitrary dimension.

See also:

Concat in developer guide

Global 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
    )

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.