Visible to Intel only — GUID: GUID-6F4E29C2-ACE5-4625-8752-418E2533503F
Visible to Intel only — GUID: GUID-6F4E29C2-ACE5-4625-8752-418E2533503F
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.