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 *output_desc, int n, const float *scales, const_mkldnn_primitive_desc_t *input_pds)
 

Detailed Description

A primitive to sum data.

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_sum_primitive_desc_create ( mkldnn_primitive_desc_t sum_primitive_desc,
const mkldnn_memory_desc_t output_desc,
int  n,
const float *  scales,
const_mkldnn_primitive_desc_t input_pds 
)

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_any format – in this case, the appropriate memory format would be chosen automatically.

Order of inputs:

Order of outputs: