| 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,
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,
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,
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,
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,
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,
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,
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,
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,
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,
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 |
|---|---|
static dnnl.rnn_flags |
dnnl.and(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
dnnl.rnn_flags |
dnnl.rnn_flags.intern() |
static dnnl.rnn_flags |
dnnl.not(dnnl.rnn_flags rhs) |
static dnnl.rnn_flags |
dnnl.or(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
static dnnl.rnn_flags |
dnnl.rnn_flags.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static dnnl.rnn_flags[] |
dnnl.rnn_flags.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
static dnnl.rnn_flags |
dnnl.xor(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
| Modifier and Type | Method and Description |
|---|---|
static dnnl.rnn_flags |
dnnl.and(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
static int[] |
dnnl.andPut(int[] lhs,
dnnl.rnn_flags rhs) |
static IntBuffer |
dnnl.andPut(IntBuffer lhs,
dnnl.rnn_flags rhs) |
static IntPointer |
dnnl.andPut(IntPointer lhs,
dnnl.rnn_flags rhs) |
static int |
dnnl.convert_to_c(dnnl.rnn_flags flags)
Converts RNN cell flags enum value from C++ API to C API type.
|
static dnnl.rnn_flags |
dnnl.not(dnnl.rnn_flags rhs) |
static dnnl.rnn_flags |
dnnl.or(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
static int[] |
dnnl.orPut(int[] lhs,
dnnl.rnn_flags rhs) |
static IntBuffer |
dnnl.orPut(IntBuffer lhs,
dnnl.rnn_flags rhs) |
static IntPointer |
dnnl.orPut(IntPointer lhs,
dnnl.rnn_flags rhs) |
static dnnl.rnn_flags |
dnnl.xor(dnnl.rnn_flags lhs,
dnnl.rnn_flags rhs) |
static int[] |
dnnl.xorPut(int[] lhs,
dnnl.rnn_flags rhs) |
static IntBuffer |
dnnl.xorPut(IntBuffer lhs,
dnnl.rnn_flags rhs) |
static IntPointer |
dnnl.xorPut(IntPointer lhs,
dnnl.rnn_flags rhs) |
Copyright © 2020. All rights reserved.