mxnet
Functions
Collaboration diagram for Softmax:

Functions

mkldnn_status_t MKLDNN_API mkldnn_softmax_forward_desc_init (mkldnn_softmax_desc_t *softmax_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, int softmax_axis)
 
mkldnn_status_t MKLDNN_API mkldnn_softmax_backward_desc_init (mkldnn_softmax_desc_t *softmax_desc, const mkldnn_memory_desc_t *diff_desc, const mkldnn_memory_desc_t *data_desc, int softmax_axis)
 

Detailed Description

A primitive to perform softmax.

See also
dev_guide_softmax in developer guide
cpp_api_softmax in cpp_api

Function Documentation

mkldnn_status_t MKLDNN_API mkldnn_softmax_backward_desc_init ( mkldnn_softmax_desc_t softmax_desc,
const mkldnn_memory_desc_t diff_desc,
const mkldnn_memory_desc_t data_desc,
int  softmax_axis 
)

Initializes a softmax_desc for backward propagation using memory descriptors diff_desc and data_desc.

Inputs:

Outputs:

mkldnn_status_t MKLDNN_API mkldnn_softmax_forward_desc_init ( mkldnn_softmax_desc_t softmax_desc,
mkldnn_prop_kind_t  prop_kind,
const mkldnn_memory_desc_t data_desc,
int  softmax_axis 
)

Initializes a softmax_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference) and memory descriptor data_desc.

Inputs:

Outputs: