Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.activation_kind(int setter) |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.alpha(float setter) |
mkldnn_rnn_cell_desc_t |
rnn_cell.desc.as_mkldnn_rnn_cell_desc_t() |
mkldnn_rnn_cell_desc_t |
rnn_cell.desc.c_rnn_cell_() |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_desc_t.cell_desc()
The RNN cell desc.
|
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.cell_kind(int setter) |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.clipping(float setter) |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.flags(int setter) |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.getPointer(long i) |
mkldnn_rnn_cell_desc_t |
mkldnn_rnn_cell_desc_t.position(long position) |
Modifier and Type | Method and Description |
---|---|
rnn_cell.desc |
rnn_cell.desc.c_rnn_cell_(mkldnn_rnn_cell_desc_t setter) |
mkldnn_rnn_desc_t |
mkldnn_rnn_desc_t.cell_desc(mkldnn_rnn_cell_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_cell_desc_init(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, and
#mkldnn_gru_linear_before_reset),
\p f (possible values are #mkldnn_eltwise_relu and
#mkldnn_eltwise_tanh), \p flags, \p alpha, and \p clipping.
|
static int |
mkldnn.mkldnn_rnn_cell_get_gates_count(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_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_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.