repository

Contents

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,
    dnnl_engine_t engine,
    const_dnnl_memory_desc_t dst_desc,
    int n,
    int concat_dimension,
    const_dnnl_memory_desc_t const* src_descs,
    const_dnnl_primitive_attr_t attr
    );

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,
    dnnl_engine_t engine,
    const_dnnl_memory_desc_t dst_desc,
    int n,
    int concat_dimension,
    const_dnnl_memory_desc_t const* src_descs,
    const_dnnl_primitive_attr_t attr
    )

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.

struct dnnl::reorder::primitive_desc struct dnnl::concat