| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Modifier and Type | Method and Description |
|---|---|
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.data_desc(dnnl_memory_desc_t setter) |
dnnl_softmax_desc_t |
softmax_forward.desc.data() |
dnnl_softmax_desc_t |
softmax_backward.desc.data() |
dnnl_softmax_desc_t |
logsoftmax_forward.desc.data() |
dnnl_softmax_desc_t |
logsoftmax_backward.desc.data() |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.diff_desc(dnnl_memory_desc_t setter) |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.getPointer(long i) |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.position(long position) |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.primitive_kind(int setter) |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.prop_kind(int setter) |
dnnl_softmax_desc_t |
dnnl_softmax_desc_t.softmax_axis(int setter) |
| Modifier and Type | Method and Description |
|---|---|
softmax_forward.desc |
softmax_forward.desc.data(dnnl_softmax_desc_t setter) |
softmax_backward.desc |
softmax_backward.desc.data(dnnl_softmax_desc_t setter) |
logsoftmax_forward.desc |
logsoftmax_forward.desc.data(dnnl_softmax_desc_t setter) |
logsoftmax_backward.desc |
logsoftmax_backward.desc.data(dnnl_softmax_desc_t setter) |
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.dnnl_logsoftmax_backward_desc_init(dnnl_softmax_desc_t logsoftmax_desc,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
int logsoftmax_axis)
Initializes a descriptor for logsoftmax backward propagation primitive.
|
static int |
dnnl.dnnl_logsoftmax_forward_desc_init(dnnl_softmax_desc_t logsoftmax_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
int logsoftmax_axis)
\} dnnl_api_softmax
|
static int |
dnnl.dnnl_softmax_backward_desc_init(dnnl_softmax_desc_t softmax_desc,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
int softmax_axis)
Initializes a descriptor for softmax backward propagation primitive.
|
static int |
dnnl.dnnl_softmax_forward_desc_init(dnnl_softmax_desc_t softmax_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
int softmax_axis)
\} dnnl_api_eltwise
|
Copyright © 2020. All rights reserved.