mxnet
Functions
Collaboration diagram for Sum:

Functions

mkldnn_status_t MKLDNN_API mkldnn_sum_primitive_desc_create (mkldnn_primitive_desc_t *sum_primitive_desc, const mkldnn_memory_desc_t *dst_mds, int n, const float *scales, const mkldnn_memory_desc_t *src_mds, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine)
 

Detailed Description

A primitive to sum data.

See also
dev_guide_sum in developer guide
cpp_api_sum in cpp_api

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_sum_primitive_desc_create ( mkldnn_primitive_desc_t sum_primitive_desc,
const mkldnn_memory_desc_t dst_mds,
int  n,
const float *  scales,
const mkldnn_memory_desc_t src_mds,
const_mkldnn_primitive_attr_t  attr,
mkldnn_engine_t  engine 
)

Creates out-of-place sum_primitive_desc for sum of n inputs multiplied by scale with resulting output_desc memory descriptor. output_desc can be NULL or specified with the mkldnn_format_kind_any format kind – in this case, the appropriate memory format would be chosen automatically.

Inputs:

Outputs: