| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Constructor and Description |
|---|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm activation,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm activation,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
dnnl.rnn_flags flags,
float alpha,
float beta)
Constructs a descriptor for a vanilla RNN forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm activation,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm activation,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
dnnl.rnn_flags flags,
float alpha,
float beta)
Constructs a descriptor for a vanilla RNN backward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for LBR GRU forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for a GRU forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for an LSTM forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for an LSTM (with or without peephole)
forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc weights_projection_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc weights_projection_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for an LSTM (with or without peephole and
with or without projection) forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for LBR GRU backward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
dnnl.rnn_flags flags)
Constructs a descriptor for a GRU backward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs an LSTM descriptor for backward propagation using \p
prop_kind, \p direction, and memory descriptors.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_weights_peephole_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_weights_peephole_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs an LSTM (with or without peephole) descriptor for
backward propagation using \p prop_kind, \p direction, and memory
descriptors.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc weights_projection_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_weights_peephole_desc,
memory.desc diff_weights_projection_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.rnn_direction direction,
memory.desc src_layer_desc,
memory.desc src_iter_desc,
memory.desc src_iter_c_desc,
memory.desc weights_layer_desc,
memory.desc weights_iter_desc,
memory.desc weights_peephole_desc,
memory.desc weights_projection_desc,
memory.desc bias_desc,
memory.desc dst_layer_desc,
memory.desc dst_iter_desc,
memory.desc dst_iter_c_desc,
memory.desc diff_src_layer_desc,
memory.desc diff_src_iter_desc,
memory.desc diff_src_iter_c_desc,
memory.desc diff_weights_layer_desc,
memory.desc diff_weights_iter_desc,
memory.desc diff_weights_peephole_desc,
memory.desc diff_weights_projection_desc,
memory.desc diff_bias_desc,
memory.desc diff_dst_layer_desc,
memory.desc diff_dst_iter_desc,
memory.desc diff_dst_iter_c_desc,
dnnl.rnn_flags flags)
Constructs an LSTM (with or without peephole and with or without
projection) descriptor for backward propagation using \p prop_kind,
\p direction, and memory descriptors.
|
| Modifier and Type | Method and Description |
|---|---|
dnnl.rnn_direction |
dnnl.rnn_direction.intern() |
static dnnl.rnn_direction |
dnnl.rnn_direction.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static dnnl.rnn_direction[] |
dnnl.rnn_direction.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.convert_to_c(dnnl.rnn_direction dir)
Converts RNN direction enum value from C++ API to C API type.
|
Copyright © 2020. All rights reserved.