| Package | Description |
|---|---|
| org.bytedeco.javacpp |
| Modifier and Type | Method and Description |
|---|---|
mkldnn.rnn_forward.desc |
mkldnn.rnn_forward.desc.data(mkldnn.mkldnn_rnn_desc_t data) |
mkldnn.rnn_backward.desc |
mkldnn.rnn_backward.desc.data(mkldnn.mkldnn_rnn_desc_t data) |
static int |
mkldnn.mkldnn_rnn_backward_desc_init(mkldnn.mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn.mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn.mkldnn_memory_desc_t src_layer_desc,
mkldnn.mkldnn_memory_desc_t src_iter_desc,
mkldnn.mkldnn_memory_desc_t weights_layer_desc,
mkldnn.mkldnn_memory_desc_t weights_iter_desc,
mkldnn.mkldnn_memory_desc_t bias_desc,
mkldnn.mkldnn_memory_desc_t dst_layer_desc,
mkldnn.mkldnn_memory_desc_t dst_iter_desc,
mkldnn.mkldnn_memory_desc_t diff_src_layer_desc,
mkldnn.mkldnn_memory_desc_t diff_src_iter_desc,
mkldnn.mkldnn_memory_desc_t diff_weights_layer_desc,
mkldnn.mkldnn_memory_desc_t diff_weights_iter_desc,
mkldnn.mkldnn_memory_desc_t diff_bias_desc,
mkldnn.mkldnn_memory_desc_t diff_dst_layer,
mkldnn.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.mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn.mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn.mkldnn_memory_desc_t src_layer_desc,
mkldnn.mkldnn_memory_desc_t src_iter_desc,
mkldnn.mkldnn_memory_desc_t weights_layer_desc,
mkldnn.mkldnn_memory_desc_t weights_iter_desc,
mkldnn.mkldnn_memory_desc_t bias_desc,
mkldnn.mkldnn_memory_desc_t dst_layer_desc,
mkldnn.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 © 2018. All rights reserved.