| Package | Description |
|---|---|
| org.bytedeco.javacpp |
| Modifier and Type | Method and Description |
|---|---|
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.data_desc(mkldnn.mkldnn_memory_desc_t data_desc) |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.softmax_forward.desc.data() |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.softmax_backward.desc.data() |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.diff_desc(mkldnn.mkldnn_memory_desc_t diff_desc) |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.position(long position) |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.primitive_kind(int primitive_kind) |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.prop_kind(int prop_kind) |
mkldnn.mkldnn_softmax_desc_t |
mkldnn.mkldnn_softmax_desc_t.softmax_axis(int softmax_axis) |
| Modifier and Type | Method and Description |
|---|---|
mkldnn.softmax_forward.desc |
mkldnn.softmax_forward.desc.data(mkldnn.mkldnn_softmax_desc_t data) |
mkldnn.softmax_backward.desc |
mkldnn.softmax_backward.desc.data(mkldnn.mkldnn_softmax_desc_t data) |
static int |
mkldnn.mkldnn_softmax_backward_desc_init(mkldnn.mkldnn_softmax_desc_t softmax_desc,
mkldnn.mkldnn_memory_desc_t diff_desc,
mkldnn.mkldnn_memory_desc_t data_desc,
int softmax_axis)
Initializes a \p softmax_desc for backward propagation using memory
descriptors \p diff_desc and \p data_desc.
|
static int |
mkldnn.mkldnn_softmax_forward_desc_init(mkldnn.mkldnn_softmax_desc_t softmax_desc,
int prop_kind,
mkldnn.mkldnn_memory_desc_t data_desc,
int softmax_axis)
Initializes a \p softmax_desc for forward propagation using \p prop_kind
(possible value are #mkldnn_forward_training or #mkldnn_forward_inference)
and memory descriptor \p data_desc.
|
Copyright © 2018. All rights reserved.