|
mxnet
|
#include <mkldnn_types.h>
Public Attributes | |
| mkldnn_primitive_kind_t | primitive_kind |
| mkldnn_prop_kind_t | prop_kind |
| mkldnn_alg_kind_t | alg_kind |
| mkldnn_memory_desc_t | data_desc |
| mkldnn_memory_desc_t | diff_data_desc |
| float | alpha |
| float | beta |
A descriptor of a element-wise operation.
| mkldnn_alg_kind_t mkldnn_eltwise_desc_t::alg_kind |
The kind of eltwise algorithm. Possible values: mkldnn_eltwise_relu, mkldnn_eltwise_tanh, mkldnn_eltwise_elu, mkldnn_eltwise_square, mkldnn_eltwise_abs, mkldnn_eltwise_sqrt, mkldnn_eltwise_linear, mkldnn_eltwise_bounded_relu, mkldnn_eltwise_soft_relu, and mkldnn_eltwise_logistic.
| float mkldnn_eltwise_desc_t::alpha |
Algorithm specific parameter. Accordance table:
alpha – negative slope, beta ignoredalpha and beta ignoredalpha – negative slope, beta ignoredalpha and beta ignoredalpha and beta ignoredalpha and beta ignoredalpha – scale, beta – shiftalpha – upper bound, beta ignoredalpha and beta ignoredalpha and beta ignored | float mkldnn_eltwise_desc_t::beta |
| mkldnn_memory_desc_t mkldnn_eltwise_desc_t::data_desc |
Source and destination memory descriptor.
| mkldnn_memory_desc_t mkldnn_eltwise_desc_t::diff_data_desc |
Source and destination gradient memory descriptor.
| mkldnn_primitive_kind_t mkldnn_eltwise_desc_t::primitive_kind |
The kind of primitive. Used for self-identifying the primitive descriptor. Must be mkldnn_eltwise.
| mkldnn_prop_kind_t mkldnn_eltwise_desc_t::prop_kind |
The kind of propagation. Possible values: mkldnn_forward_training, mkldnn_forward_inference, mkldnn_backward, and mkldnn_backward_data.
1.8.11