Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.axis(int setter) |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.data_desc(mkldnn_memory_desc_t setter) |
mkldnn_shuffle_desc_t |
shuffle_forward.desc.data() |
mkldnn_shuffle_desc_t |
shuffle_backward.desc.data() |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.getPointer(long i) |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.group_size(int setter) |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.position(long position) |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.primitive_kind(int setter) |
mkldnn_shuffle_desc_t |
mkldnn_shuffle_desc_t.prop_kind(int setter) |
Modifier and Type | Method and Description |
---|---|
shuffle_forward.desc |
shuffle_forward.desc.data(mkldnn_shuffle_desc_t setter) |
shuffle_backward.desc |
shuffle_backward.desc.data(mkldnn_shuffle_desc_t setter) |
Modifier and Type | Method and Description |
---|---|
static int |
mkldnn.mkldnn_shuffle_backward_desc_init(mkldnn_shuffle_desc_t shuffle_desc,
mkldnn_memory_desc_t diff_data_desc,
int axis,
int group_size)
Initializes a \p shuffle_desc for backward propagation using memory
descriptor \p diff_data_desc, \p axis, and \p group_size.
|
static int |
mkldnn.mkldnn_shuffle_forward_desc_init(mkldnn_shuffle_desc_t shuffle_desc,
int prop_kind,
mkldnn_memory_desc_t data_desc,
int axis,
int group_size)
Initializes a \p shuffle_desc for forward propagation using \p prop_kind,
memory descriptor \p data_desc, \p axis, and \p group_size.
|
Copyright © 2022. All rights reserved.