Binary

A primitive to perform tensor operations over two tensors. More...

Functions
dnnl_status_t DNNL_API	dnnl_binary_desc_init (dnnl_binary_desc_t *binary_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *src0_desc, const dnnl_memory_desc_t *src1_desc, const dnnl_memory_desc_t *dst_desc)
	Initializes a binary descriptor binary_desc, alg_kind (possible values are dnnl_binary_add and dnnl_binary_mul), and memory descriptors. More...

Detailed Description

A primitive to perform tensor operations over two tensors.

See also

Binary in developer guide

Binary in C++ API

Function Documentation

dnnl_binary_desc_init()

dnnl_status_t DNNL_API dnnl_binary_desc_init	(	dnnl_binary_desc_t *	binary_desc,
		dnnl_alg_kind_t	alg_kind,
		const dnnl_memory_desc_t *	src0_desc,
		const dnnl_memory_desc_t *	src1_desc,
		const dnnl_memory_desc_t *	dst_desc
	)

Initializes a binary descriptor binary_desc, alg_kind (possible values are dnnl_binary_add and dnnl_binary_mul), and memory descriptors.

Note

Memory descriptor dst_desc can have format_kind set to dnnl_format_kind_any.

Both memory descriptors must have the same number of dimensions. Element broadcasting is supported for memory descriptor src1_desc and are applied to dimensions for which @ src1_desc has size equal to 1.

Inputs:

• src0 (dnnl_query_src_md, 0)
• src1 (dnnl_query_src_md, 1)

Outputs:

• dst (dnnl_query_dst_md, 0)