A primitive to perform tensor operations over two tensors. More...
Classes | |
struct | dnnl::binary |
Elementwise binary operator primitive. More... | |
struct | dnnl_binary_desc_t |
A descriptor of a binary operation. 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 descriptor for a binary primitive. More... | |
A primitive to perform tensor operations over two tensors.
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 descriptor for a binary primitive.
dst_desc
is allowed to be initialized with dnnl_format_tag_any or with format_kind set to dnnl_format_kind_any.src1_desc
and are applied to @ src1_desc dimensions that have size equal to 1.binary_desc | Output descriptor for a binary primitive. |
alg_kind | Algorithm kind. Valid values are dnnl_binary_add, dnnl_binary_mul, dnnl_binary_max, dnnl_binary_min, dnnl_binary_div and dnnl_binary_sub. |
src0_desc | Source 0 memory descriptor. |
src1_desc | Source 1 memory descriptor. |
dst_desc | Destination memory descriptor. |