A descriptor of reduction operation. More...
#include <dnnl_types.h>
Public Attributes | |
| dnnl_primitive_kind_t | primitive_kind |
| dnnl_alg_kind_t | alg_kind |
| dnnl_memory_desc_t | src_desc |
| Source memory descriptor. More... | |
| dnnl_memory_desc_t | dst_desc |
| Destination memory descriptor. More... | |
| float | p |
| float | eps |
Detailed Description
A descriptor of reduction operation.
Member Data Documentation
◆ alg_kind
| dnnl_alg_kind_t dnnl_reduction_desc_t::alg_kind |
The kind of reduction algorithm. Possible values: dnnl_reduction_max, dnnl_reduction_min, dnnl_reduction_sum, dnnl_reduction_mul, dnnl_reduction_mean, dnnl_reduction_norm_lp_max, dnnl_reduction_norm_lp_sum, dnnl_reduction_norm_lp_power_p_max, dnnl_reduction_norm_lp_power_p_sum.
◆ dst_desc
| dnnl_memory_desc_t dnnl_reduction_desc_t::dst_desc |
Destination memory descriptor.
◆ eps
| float dnnl_reduction_desc_t::eps |
◆ p
| float dnnl_reduction_desc_t::p |
Algorithm specific parameters. Accordance table: dnnl_reduction_max: p and eps are ignored dnnl_reduction_min: p and eps are ignored dnnl_reduction_norm_lp_max: p – power, eps – epsilon dnnl_reduction_norm_lp_sum: p – power, eps – epsilon dnnl_reduction_norm_lp_power_p_max: p – power, eps – epsilon dnnl_reduction_norm_lp_power_p_sum: p – power, eps – epsilon dnnl_reduction_sum: p and eps are ignored dnnl_reduction_mul: p and eps are ignored dnnl_reduction_mean: p and eps are ignored
◆ primitive_kind
| dnnl_primitive_kind_t dnnl_reduction_desc_t::primitive_kind |
The kind of primitive. Used for self-identifying the primitive descriptor. Must be dnnl_reduction.
◆ src_desc
| dnnl_memory_desc_t dnnl_reduction_desc_t::src_desc |
Source memory descriptor.
The documentation for this struct was generated from the following file:
- /work/mxnet/include/mkldnn/oneapi/dnnl/dnnl_types.h
