Sum

A primitive to sum multiple tensors. More...

Classes
struct	dnnl::sum
	Out-of-place summation (sum) primitive. More...

Functions
dnnl_status_t DNNL_API	dnnl_sum_primitive_desc_create (dnnl_primitive_desc_t *sum_primitive_desc, const dnnl_memory_desc_t *dst_desc, int n, const float *scales, 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) sum primitive. More...

Detailed Description

A primitive to sum multiple tensors.

See also

Sum in developer guide

Function Documentation

dnnl_sum_primitive_desc_create()

dnnl_status_t DNNL_API dnnl_sum_primitive_desc_create	(	dnnl_primitive_desc_t *	sum_primitive_desc,
		const dnnl_memory_desc_t *	dst_desc,
		int	n,
		const float *	scales,
		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) sum primitive.

Inputs:

• src[0] (dnnl_query_src_md, 0)
• src[1] (dnnl_query_src_md, 1)
• ...
• src[n - 1] (dnnl_query_src_md, n - 1)

Outputs:

• dst (dnnl_query_dst_md, 0)

Parameters

sum_primitive_desc	Output primitive descriptor.
dst_desc	Destination memory descriptor.
n	Number of source parameters.
scales	Vector of scales to multiply data in each source memory by.
src_descs	Array of source memory descriptors having 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.