| 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.algorithm aalgorithm,
float[] factors,
memory.desc src_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
float[] factors,
memory.desc src_desc,
memory.desc dst_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
FloatBuffer factors,
memory.desc src_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
FloatBuffer factors,
memory.desc src_desc,
memory.desc dst_desc) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
FloatPointer factors,
memory.desc src_desc)
Constructs a descriptor for a resampling forward propagation
primitive using source memory descriptor and factors.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
FloatPointer factors,
memory.desc src_desc,
memory.desc dst_desc)
Constructs a descriptor for a resampling forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc data_desc,
float alpha,
float beta)
Constructs a descriptor for an elementwise forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc data_desc,
long local_size,
float alpha,
float beta) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc data_desc,
long local_size,
float alpha,
float beta,
float k)
Constructs a descriptor for a LRN forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc dst_desc)
Constructs a descriptor for a resampling forward propagation
primitive using source and destination memory descriptors.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc dst_desc,
long[] strides,
long[] kernel,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer kernel,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer kernel,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for pooling forward propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a deconvolution forward propagation
primitive without bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a convolution forward propagation
primitive without bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a dilated deconvolution forward
propagation primitive without bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a dilated convolution forward
propagation primitive without bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a deconvolution forward propagation
primitive with bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a convolution forward propagation
primitive with bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a dilated deconvolution forward
propagation primitive with bias.
|
desc(dnnl.prop_kind aprop_kind,
dnnl.algorithm aalgorithm,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Constructs a descriptor for a dilated convolution forward
propagation primitive with bias.
|
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.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a descriptor for layer normalization forward
propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a batch normalization descriptor for forward
propagation.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
int softmax_axis)
Constructs a descriptor for a softmax forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
int logsoftmax_axis)
Constructs a descriptor for a logsoftmax forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
int axis,
int group_size)
Constructs a descriptor for a shuffle forward propagation
primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc data_desc,
memory.desc stat_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a descriptor for layer normalization forward
propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc diff_data_desc,
memory.desc data_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a descriptor for layer normalization backward
propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc diff_data_desc,
memory.desc data_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a batch normalization descriptor for backward
propagation.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc dst_desc)
Constructs a descriptor for an inner product forward propagation
primitive without bias.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc diff_data_desc,
memory.desc data_desc,
memory.desc stat_desc,
float epsilon,
dnnl.normalization_flags flags)
Constructs a descriptor for layer normalization backward
propagation primitive.
|
desc(dnnl.prop_kind aprop_kind,
memory.desc src_desc,
memory.desc weights_desc,
memory.desc bias_desc,
memory.desc dst_desc)
Constructs a descriptor for an inner product forward propagation
primitive with bias.
|
rnn_primitive_desc_base(dnnl_primitive_desc pd,
dnnl.prop_kind aprop_kind,
dnnl.algorithm cell_kind)
Constructs an RNN primitive descriptor base from a C API primitive
descriptor while checking that it actually describes the expected
primitive by comparing propagation and primitive kinds.
|
| Modifier and Type | Method and Description |
|---|---|
dnnl.prop_kind |
dnnl.prop_kind.intern() |
static dnnl.prop_kind |
dnnl.prop_kind.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static dnnl.prop_kind[] |
dnnl.prop_kind.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.prop_kind akind)
Converts propagation kind enum value from C++ API to C API type.
|
Copyright © 2020. All rights reserved.