mxnet
|
Typedefs | |
typedef dnnl_softmax_desc_t | dnnl_logsoftmax_desc_t |
Functions | |
dnnl_status_t DNNL_API | dnnl_logsoftmax_forward_desc_init (dnnl_logsoftmax_desc_t *logsoftmax_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t *data_desc, int logsoftmax_axis) |
dnnl_status_t DNNL_API | dnnl_logsoftmax_backward_desc_init (dnnl_logsoftmax_desc_t *logsoftmax_desc, const dnnl_memory_desc_t *diff_data_desc, const dnnl_memory_desc_t *data_desc, int logsoftmax_axis) |
A descriptor of a LogSoftmax operation. An alias of Softmax structure, but primitive_kind must be dnnl_logsoftmax.
dnnl_status_t DNNL_API dnnl_logsoftmax_backward_desc_init | ( | dnnl_logsoftmax_desc_t * | logsoftmax_desc, |
const dnnl_memory_desc_t * | diff_data_desc, | ||
const dnnl_memory_desc_t * | data_desc, | ||
int | logsoftmax_axis | ||
) |
Initializes a descriptor for logsoftmax backward propagation primitive.
logsoftmax_desc | Output descriptor for a logsoftmax primitive. |
diff_data_desc | Diff source and diff destination memory descriptors. |
data_desc | Destination memory descriptor. |
logsoftmax_axis | Axis over which softmax is computed. |
dnnl_status_t DNNL_API dnnl_logsoftmax_forward_desc_init | ( | dnnl_logsoftmax_desc_t * | logsoftmax_desc, |
dnnl_prop_kind_t | prop_kind, | ||
const dnnl_memory_desc_t * | data_desc, | ||
int | logsoftmax_axis | ||
) |
Initializes a descriptor for logsoftmax forward propagation primitive.
logsoftmax_desc | Output descriptor for a logsoftmax primitive. |
prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
data_desc | Source and destination memory descriptor. |
logsoftmax_axis | Axis over which logsoftmax is computed. |