mxnet
|
Classes | |
struct | dnnl_softmax_v2_desc_t |
A descriptor of a Softmax operation. More... | |
Functions | |
dnnl_status_t DNNL_API | dnnl_softmax_v2_forward_desc_init (dnnl_softmax_v2_desc_t *softmax_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *src_desc, const dnnl_memory_desc_t *dst_desc, int softmax_axis) |
dnnl_status_t DNNL_API | dnnl_softmax_v2_backward_desc_init (dnnl_softmax_v2_desc_t *softmax_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *diff_src_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_memory_desc_t *dst_desc, int softmax_axis) |
dnnl_status_t DNNL_API dnnl_softmax_v2_backward_desc_init | ( | dnnl_softmax_v2_desc_t * | softmax_desc, |
dnnl_alg_kind_t | alg_kind, | ||
const dnnl_memory_desc_t * | diff_src_desc, | ||
const dnnl_memory_desc_t * | diff_dst_desc, | ||
const dnnl_memory_desc_t * | dst_desc, | ||
int | softmax_axis | ||
) |
Initializes a descriptor for softmax v2 backward propagation primitive.
softmax_desc | Output descriptor for a softmax primitive. |
alg_kind | Softmax algorithm kind: either dnnl_softmax_accurate, or dnnl_softmax_log. |
diff_src_desc | Diff source memory descriptor. |
diff_dst_desc | Diff destination memory descriptor. |
dst_desc | Destination memory descriptor. |
softmax_axis | Axis over which softmax is computed. |
dnnl_status_t DNNL_API dnnl_softmax_v2_forward_desc_init | ( | dnnl_softmax_v2_desc_t * | softmax_desc, |
dnnl_prop_kind_t | prop_kind, | ||
dnnl_alg_kind_t | alg_kind, | ||
const dnnl_memory_desc_t * | src_desc, | ||
const dnnl_memory_desc_t * | dst_desc, | ||
int | softmax_axis | ||
) |
Initializes a descriptor for softmax v2 forward propagation primitive.
softmax_desc | Output descriptor for a softmax primitive. |
prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
alg_kind | Softmax algorithm kind: either dnnl_softmax_accurate, or dnnl_softmax_log. |
src_desc | Source memory descriptor. |
dst_desc | Destination memory descriptor. |
softmax_axis | Axis over which softmax is computed. |