A primitive to sum data. More...
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) |
Creates out-of-place sum_primitive_desc for sum of n inputs multiplied by scale with resulting output_desc memory descriptor. More... | |
A primitive to sum data.
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:
n
- 1 (mkldnn_query_src_md, n
- 1)Outputs: