mxnet
|
Classes | |
struct | dnnl_eltwise_desc_t |
A descriptor of a element-wise operation. More... | |
Functions | |
dnnl_status_t DNNL_API | dnnl_eltwise_forward_desc_init (dnnl_eltwise_desc_t *eltwise_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *data_desc, float alpha, float beta) |
dnnl_status_t DNNL_API | dnnl_eltwise_backward_desc_init (dnnl_eltwise_desc_t *eltwise_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *diff_data_desc, const dnnl_memory_desc_t *data_desc, float alpha, float beta) |
dnnl_status_t DNNL_API dnnl_eltwise_backward_desc_init | ( | dnnl_eltwise_desc_t * | eltwise_desc, |
dnnl_alg_kind_t | alg_kind, | ||
const dnnl_memory_desc_t * | diff_data_desc, | ||
const dnnl_memory_desc_t * | data_desc, | ||
float | alpha, | ||
float | beta | ||
) |
Initializes a descriptor for eltwise backward propagation primitive.
eltwise_desc | Output descriptor for an eltwise primitive. |
alg_kind | Elementwise algorithm kind. |
diff_data_desc | Diff source and diff destination memory descriptors. |
data_desc | Source and destination memory descriptor. |
alpha | The alpha parameter for the elementwise operation. Specific meaning depends on the algorithm. |
beta | The beta parameter for the elementwise operation. Specific meaning depends on the algorithm. |
dnnl_status_t DNNL_API dnnl_eltwise_forward_desc_init | ( | dnnl_eltwise_desc_t * | eltwise_desc, |
dnnl_prop_kind_t | prop_kind, | ||
dnnl_alg_kind_t | alg_kind, | ||
const dnnl_memory_desc_t * | data_desc, | ||
float | alpha, | ||
float | beta | ||
) |
Initializes a descriptor for eltwise forward propagation primitive.
eltwise_desc | Output descriptor for an eltwise primitive. |
prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
alg_kind | Elementwise algorithm kind. |
data_desc | Source and destination memory descriptor. |
alpha | The alpha parameter for the elementwise operation. Specific meaning depends on the algorithm. |
beta | The beta parameter for the elementwise operation. Specific meaning depends on the algorithm. |