A primitive to perform softmax. More...
Classes | |
| struct | dnnl::softmax_forward |
| Softmax forward propagation primitive. More... | |
| struct | dnnl::softmax_backward |
| Softmax backward propagation primitive. More... | |
| struct | dnnl_softmax_desc_t |
| A descriptor of a Softmax operation. More... | |
Functions | |
| dnnl_status_t DNNL_API | dnnl_softmax_forward_desc_init (dnnl_softmax_desc_t *softmax_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t *data_desc, int softmax_axis) |
| Initializes a descriptor for softmax forward propagation primitive. More... | |
| dnnl_status_t DNNL_API | dnnl_softmax_backward_desc_init (dnnl_softmax_desc_t *softmax_desc, const dnnl_memory_desc_t *diff_data_desc, const dnnl_memory_desc_t *data_desc, int softmax_axis) |
| Initializes a descriptor for softmax backward propagation primitive. More... | |
A primitive to perform softmax.
| dnnl_status_t DNNL_API dnnl_softmax_forward_desc_init | ( | dnnl_softmax_desc_t * | softmax_desc, |
| dnnl_prop_kind_t | prop_kind, | ||
| const dnnl_memory_desc_t * | data_desc, | ||
| int | softmax_axis | ||
| ) |
Initializes a descriptor for softmax forward propagation primitive.
Inputs:
Outputs:
| softmax_desc | Output descriptor for a softmax primitive. |
| prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
| data_desc | Source and destination memory descriptor. |
| softmax_axis | Axis over which softmax is computed. |
| dnnl_status_t DNNL_API dnnl_softmax_backward_desc_init | ( | dnnl_softmax_desc_t * | softmax_desc, |
| const dnnl_memory_desc_t * | diff_data_desc, | ||
| const dnnl_memory_desc_t * | data_desc, | ||
| int | softmax_axis | ||
| ) |
Initializes a descriptor for softmax backward propagation primitive.
Inputs:
Outputs:
| softmax_desc | Output descriptor for a softmax primitive. |
| diff_data_desc | Diff source and diff destination memory descriptors. |
| data_desc | Destination memory descriptor. |
| softmax_axis | Axis over which softmax is computed. |