| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Modifier and Type | Method and Description |
|---|---|
vanilla_rnn_forward.desc |
vanilla_rnn_forward.desc.data(dnnl_rnn_desc_t setter) |
vanilla_rnn_backward.desc |
vanilla_rnn_backward.desc.data(dnnl_rnn_desc_t setter) |
lstm_forward.desc |
lstm_forward.desc.data(dnnl_rnn_desc_t setter) |
lstm_backward.desc |
lstm_backward.desc.data(dnnl_rnn_desc_t setter) |
lbr_gru_forward.desc |
lbr_gru_forward.desc.data(dnnl_rnn_desc_t setter) |
lbr_gru_backward.desc |
lbr_gru_backward.desc.data(dnnl_rnn_desc_t setter) |
gru_forward.desc |
gru_forward.desc.data(dnnl_rnn_desc_t setter) |
gru_backward.desc |
gru_backward.desc.data(dnnl_rnn_desc_t setter) |
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.dnnl_gru_backward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
int flags)
Initializes a descriptor for GRU backward propagation primitive.
|
static int |
dnnl.dnnl_gru_forward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
int flags)
Initializes a descriptor for GRU forward propagation primitive.
|
static int |
dnnl.dnnl_lbr_gru_backward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
int flags)
Initializes a descriptor for LBR GRU backward propagation primitive.
|
static int |
dnnl.dnnl_lbr_gru_forward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
int flags)
Initializes a descriptor for LBR GRU forward propagation primitive.
|
static int |
dnnl.dnnl_lstm_backward_desc_init_v2(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t weights_peephole_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_src_iter_c_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_weights_peephole_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
dnnl_memory_desc_t diff_dst_iter_c_desc,
int flags)
Initializes a descriptor for an LSTM (with or without peephole) backward
propagation primitive.
|
static int |
dnnl.dnnl_lstm_backward_desc_init_v3(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t weights_peephole_desc,
dnnl_memory_desc_t weights_projection_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_src_iter_c_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_weights_peephole_desc,
dnnl_memory_desc_t diff_weights_projection_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
dnnl_memory_desc_t diff_dst_iter_c_desc,
int flags)
Initializes a descriptor for an LSTM (with or without peephole and with or
with out recurrent projection layer) backward propagation primitive.
|
static int |
dnnl.dnnl_lstm_backward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_src_iter_c_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
dnnl_memory_desc_t diff_dst_iter_c_desc,
int flags)
Initializes a descriptor for an LSTM backward propagation primitive.
|
static int |
dnnl.dnnl_lstm_forward_desc_init_v2(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t weights_peephole_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
int flags)
Initializes a descriptor for an LSTM (with or without peephole) forward
propagation primitive.
|
static int |
dnnl.dnnl_lstm_forward_desc_init_v3(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t weights_peephole_desc,
dnnl_memory_desc_t weights_projection_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
int flags)
Initializes a descriptor for an LSTM (with or without peephole and with
or without recurrent projection layer) forward propagation primitive.
|
static int |
dnnl.dnnl_lstm_forward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t src_iter_c_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t dst_iter_c_desc,
int flags)
Initializes a descriptor for LSTM forward propagation primitive.
|
static int |
dnnl.dnnl_vanilla_rnn_backward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int activation,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
dnnl_memory_desc_t diff_src_layer_desc,
dnnl_memory_desc_t diff_src_iter_desc,
dnnl_memory_desc_t diff_weights_layer_desc,
dnnl_memory_desc_t diff_weights_iter_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_layer_desc,
dnnl_memory_desc_t diff_dst_iter_desc,
int flags,
float alpha,
float beta)
Initializes a descriptor for vanilla RNN backward propagation primitive.
|
static int |
dnnl.dnnl_vanilla_rnn_forward_desc_init(dnnl_rnn_desc_t rnn_desc,
int prop_kind,
int activation,
int direction,
dnnl_memory_desc_t src_layer_desc,
dnnl_memory_desc_t src_iter_desc,
dnnl_memory_desc_t weights_layer_desc,
dnnl_memory_desc_t weights_iter_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_layer_desc,
dnnl_memory_desc_t dst_iter_desc,
int flags,
float alpha,
float beta)
\} dnnl_api_attributes
|
Copyright © 2020. All rights reserved.