| Package | Description |
|---|---|
| org.bytedeco.javacpp |
| Modifier and Type | Method and Description |
|---|---|
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.activation_kind(int activation_kind) |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.alpha(float alpha) |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.rnn_cell.desc.as_mkldnn_rnn_cell_desc_t() |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.rnn_cell.desc.c_rnn_cell_() |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_desc_t.cell_desc()
The RNN cell desc.
|
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.cell_kind(int cell_kind) |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.clipping(float clipping) |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.flags(int flags) |
mkldnn.mkldnn_rnn_cell_desc_t |
mkldnn.mkldnn_rnn_cell_desc_t.position(long position) |
| Modifier and Type | Method and Description |
|---|---|
mkldnn.rnn_cell.desc |
mkldnn.rnn_cell.desc.c_rnn_cell_(mkldnn.mkldnn_rnn_cell_desc_t c_rnn_cell_) |
mkldnn.mkldnn_rnn_desc_t |
mkldnn.mkldnn_rnn_desc_t.cell_desc(mkldnn.mkldnn_rnn_cell_desc_t cell_desc) |
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_cell_desc_init(mkldnn.mkldnn_rnn_cell_desc_t rnn_cell_desc,
int kind,
int f,
int flags,
float alpha,
float clipping)
Initializes a recurrent cell descriptor \p rnn_cell_desc
using \p rnn_cell_desc, \p kind (possible values are
#mkldnn_vanilla_rnn, #mkldnn_vanilla_lstm, #mkldnn_vanilla_gru,
#mkldnn_gru_linear_before_reset),
\p f (possible values are #mkldnn_eltwise_relu,
#mkldnn_eltwise_tanh), \p flags, \p alpha, and \p clipping.
|
static int |
mkldnn.mkldnn_rnn_cell_get_gates_count(mkldnn.mkldnn_rnn_cell_desc_t rnn_cell_desc)
Returns the number of gates of a particular \p rnn_cell_desc.
|
static int |
mkldnn.mkldnn_rnn_cell_get_states_count(mkldnn.mkldnn_rnn_cell_desc_t rnn_cell_desc)
Returns the number of states of a particular \p rnn_cell_desc.
|
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.