Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
rnn_forward.desc |
rnn_forward.desc.data(mkldnn_rnn_desc_t setter) |
rnn_backward.desc |
rnn_backward.desc.data(mkldnn_rnn_desc_t setter) |
Modifier and Type | Method and Description |
---|---|
static int |
mkldnn.mkldnn_rnn_backward_desc_init(mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn_memory_desc_t src_layer_desc,
mkldnn_memory_desc_t src_iter_desc,
mkldnn_memory_desc_t weights_layer_desc,
mkldnn_memory_desc_t weights_iter_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_layer_desc,
mkldnn_memory_desc_t dst_iter_desc,
mkldnn_memory_desc_t diff_src_layer_desc,
mkldnn_memory_desc_t diff_src_iter_desc,
mkldnn_memory_desc_t diff_weights_layer_desc,
mkldnn_memory_desc_t diff_weights_iter_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_layer,
mkldnn_memory_desc_t diff_dst_iter_desc)
Initializes a rnn descriptor \p rnn_desc for backward propagation
using \p prop_kind, \p rnn_cell_desc, \p direction, and memory descriptors.
|
static int |
mkldnn.mkldnn_rnn_forward_desc_init(mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn_memory_desc_t src_layer_desc,
mkldnn_memory_desc_t src_iter_desc,
mkldnn_memory_desc_t weights_layer_desc,
mkldnn_memory_desc_t weights_iter_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_layer_desc,
mkldnn_memory_desc_t dst_iter_desc)
Initializes a rnn descriptor \p rnn_desc for forward propagation
using \p prop_kind, \p rnn_cell_desc, \p direction, and memory descriptors.
|
Copyright © 2022. All rights reserved.