| Package | Description |
|---|---|
| org.bytedeco.mkldnn | |
| org.bytedeco.mkldnn.global |
| Modifier and Type | Method and Description |
|---|---|
inner_product_forward.desc |
inner_product_forward.desc.data(mkldnn_inner_product_desc_t setter) |
inner_product_backward_weights.desc |
inner_product_backward_weights.desc.data(mkldnn_inner_product_desc_t setter) |
inner_product_backward_data.desc |
inner_product_backward_data.desc.data(mkldnn_inner_product_desc_t setter) |
| Modifier and Type | Method and Description |
|---|---|
static int |
mkldnn.mkldnn_inner_product_backward_data_desc_init(mkldnn_inner_product_desc_t ip_desc,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc)
Initializes an inner product descriptor \p ip_desc for backward propagation
with respect to data using memory descriptors.
|
static int |
mkldnn.mkldnn_inner_product_backward_weights_desc_init(mkldnn_inner_product_desc_t ip_desc,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc)
Initializes an inner product descriptor \p ip_desc for backward propagation
with respect to weights using memory descriptors.
|
static int |
mkldnn.mkldnn_inner_product_forward_desc_init(mkldnn_inner_product_desc_t ip_desc,
int prop_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc)
Initializes an inner product descriptor \p ip_desc for forward propagation
using \p prop_kind (possible values are #mkldnn_forward_training and
#mkldnn_forward_inference) and memory descriptors.
|
Copyright © 2019. All rights reserved.