A primitive to concatenate data by arbitrary dimension. More...
Classes | |
| struct | dnnl::concat |
| Tensor concatenation (concat) primitive. More... | |
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. More... | |
A primitive to concatenate data by arbitrary dimension.
| 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.
Inputs:
n - 1] (dnnl_query_src_md, n - 1)Outputs:
| 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. |