| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Modifier and Type | Method and Description |
|---|---|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.bias_desc()
Bias memory descriptor.
|
dnnl_memory_desc_t |
dnnl_matmul_desc_t.bias_desc()
Bias memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.bias_desc()
Bias memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.bias_desc()
Bias memory descriptor.
|
dnnl_memory_desc_t |
dnnl_softmax_desc_t.data_desc()
Source and destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_shuffle_desc_t.data_desc()
Source and destination memory descriptor,
and source and destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_lrn_desc_t.data_desc()
Source and destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_layer_normalization_desc_t.data_desc()
Source and destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_eltwise_desc_t.data_desc()
Source and destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_batch_normalization_desc_t.data_desc()
Source and destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_layer_normalization_desc_t.data_scaleshift_desc()
Scale and shift data and gradient memory descriptors.
|
dnnl_memory_desc_t |
dnnl_batch_normalization_desc_t.data_scaleshift_desc()
Scale and shift data and gradient memory descriptors.
|
dnnl_memory_desc_t |
dnnl_memory_desc_t.data_type(int setter) |
dnnl_memory_desc_t |
memory.desc.data()
The underlying C API data structure.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_lrn_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_layer_normalization_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_eltwise_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_batch_normalization_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_layer_normalization_desc_t.diff_data_scaleshift_desc() |
dnnl_memory_desc_t |
dnnl_batch_normalization_desc_t.diff_data_scaleshift_desc() |
dnnl_memory_desc_t |
dnnl_softmax_desc_t.diff_desc()
Source and Destination of gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_resampling_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_pooling_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_dst_iter_c_desc()
Destination gradient iteration memory descriptor for cell state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_dst_iter_desc()
Destination gradient iteration memory descriptor for hidden state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_dst_layer_desc()
Destination gradient layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_resampling_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_pooling_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_src_iter_c_desc()
Source gradient iter memory descriptor for cell state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_src_iter_desc()
Source gradient iter memory descriptor for hidden state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_src_layer_desc()
Source gradient layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.diff_weights_desc()
Weights gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.diff_weights_desc()
Weights gradient memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_weights_iter_desc()
Weights gradient iter memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_weights_layer_desc()
Weights gradient layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_weights_peephole_desc()
Weights gradient peephole memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.diff_weights_projection_desc()
Weights gradient projection memory descriptor.
|
dnnl_memory_desc_t |
dnnl_resampling_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_pooling_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_matmul_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_binary_desc_t.dst_desc()
Destination memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.dst_iter_c_desc()
Destination iter memory descriptor for cell state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.dst_iter_desc()
Destination iter memory descriptor for hidden state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.dst_layer_desc()
Destination layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_memory_desc_t.extra(dnnl_memory_extra_desc_t setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.format_desc_blocking(dnnl_blocking_desc_t setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.format_desc_rnn_packed_desc(dnnl_rnn_packed_desc_t setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.format_desc_wino_desc(dnnl_wino_desc_t setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.format_kind(int setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.getPointer(long i) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.ndims(int setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.offset0(long setter) |
dnnl_memory_desc_t |
dnnl_memory_desc_t.position(long position) |
dnnl_memory_desc_t |
dnnl_resampling_desc_t.src_desc()
Source memory descriptor.
|
dnnl_memory_desc_t |
dnnl_pooling_desc_t.src_desc()
Source memory descriptor.
|
dnnl_memory_desc_t |
dnnl_matmul_desc_t.src_desc()
Source memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.src_desc()
Source memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.src_desc()
Source memory descriptor.
|
dnnl_memory_desc_t |
dnnl_binary_desc_t.src_desc() |
dnnl_memory_desc_t |
dnnl_binary_desc_t.src_desc(int i)
Source memory descriptors.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.src_iter_c_desc()
Source iteration memory descriptor for cell state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.src_iter_desc()
Source iteration memory descriptor for hidden state.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.src_layer_desc()
Source layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_layer_normalization_desc_t.stat_desc()
Mean and variance data memory descriptors.
|
dnnl_memory_desc_t |
dnnl_batch_normalization_desc_t.stat_desc()
Statistics memory descriptor.
|
dnnl_memory_desc_t |
dnnl_matmul_desc_t.weights_desc()
Weights memory descriptor.
|
dnnl_memory_desc_t |
dnnl_inner_product_desc_t.weights_desc()
Weights memory descriptor.
|
dnnl_memory_desc_t |
dnnl_convolution_desc_t.weights_desc()
Weights memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.weights_iter_desc()
Weights iteration memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.weights_layer_desc()
Weights layer memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.weights_peephole_desc()
Weights peephole memory descriptor.
|
dnnl_memory_desc_t |
dnnl_rnn_desc_t.weights_projection_desc()
Weights projection memory descriptor.
|
| Constructor and Description |
|---|
desc(dnnl_memory_desc_t data)
Constructs a memory descriptor from a C API data structure.
|
| Modifier and Type | Method and Description |
|---|---|
static dnnl_memory_desc_t |
dnnl.convert_to_c(memory.desc mems)
\} dnnl_api_reorder
|
static dnnl_memory_desc_t |
dnnl.dnnl_primitive_desc_query_md(dnnl_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for a memory descriptor.
|
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.dnnl_batch_normalization_backward_desc_init(dnnl_batch_normalization_desc_t bnrm_desc,
int prop_kind,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
float epsilon,
int flags)
Initializes a descriptor for a batch normalization backward propagation
primitive.
|
static int |
dnnl.dnnl_batch_normalization_forward_desc_init(dnnl_batch_normalization_desc_t bnrm_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
float epsilon,
int flags)
\} dnnl_api_lrn
|
static int |
dnnl.dnnl_binary_desc_init(dnnl_binary_desc_t binary_desc,
int alg_kind,
dnnl_memory_desc_t src0_desc,
dnnl_memory_desc_t src1_desc,
dnnl_memory_desc_t dst_desc)
\} dnnl_api_sum
|
static int |
dnnl.dnnl_concat_primitive_desc_create(dnnl_primitive_desc concat_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
int concat_dimension,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine)
\} dnnl_api_reorder
|
static int |
dnnl.dnnl_concat_primitive_desc_create(PointerPointer concat_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
int concat_dimension,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a convolution backward propagation primitive.
|
static int |
dnnl.dnnl_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a convolution weights gradient primitive.
|
static int |
dnnl.dnnl_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
\} dnnl_api_binary
|
static int |
dnnl.dnnl_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a deconvolution backward propagation primitive.
|
static int |
dnnl.dnnl_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a deconvolution weights gradient primitive.
|
static int |
dnnl.dnnl_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
long[] strides,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongBuffer strides,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongPointer strides,
LongPointer padding_l,
LongPointer padding_r)
\} dnnl_api_convolution
|
static int |
dnnl.dnnl_dilated_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_convolution_backward_data_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated convolution backward propagation
primitive.
|
static int |
dnnl.dnnl_dilated_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_convolution_backward_weights_desc_init(dnnl_convolution_desc_t conv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated convolution weights gradient
primitive.
|
static int |
dnnl.dnnl_dilated_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_convolution_forward_desc_init(dnnl_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated convolution forward propagation
primitive.
|
static int |
dnnl.dnnl_dilated_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_backward_data_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated deconvolution backward propagation
primitive.
|
static int |
dnnl.dnnl_dilated_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_backward_weights_desc_init(dnnl_convolution_desc_t deconv_desc,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated deconvolution weights gradient
primitive.
|
static int |
dnnl.dnnl_dilated_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
long[] strides,
long[] dilates,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongBuffer strides,
LongBuffer dilates,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_dilated_deconvolution_forward_desc_init(dnnl_convolution_desc_t deconv_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc,
LongPointer strides,
LongPointer dilates,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for a dilated deconvolution forward propagation
primitive.
|
static int |
dnnl.dnnl_eltwise_backward_desc_init(dnnl_eltwise_desc_t eltwise_desc,
int alg_kind,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
float alpha,
float beta)
Initializes a descriptor for eltwise backward propagation primitive.
|
static int |
dnnl.dnnl_eltwise_forward_desc_init(dnnl_eltwise_desc_t eltwise_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t data_desc,
float alpha,
float beta)
\} dnnl_api_shuffle
|
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_inner_product_backward_data_desc_init(dnnl_inner_product_desc_t ip_desc,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t diff_dst_desc)
Initializes descriptor for inner product backward propagation.
|
static int |
dnnl.dnnl_inner_product_backward_weights_desc_init(dnnl_inner_product_desc_t ip_desc,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t diff_weights_desc,
dnnl_memory_desc_t diff_bias_desc,
dnnl_memory_desc_t diff_dst_desc)
Initializes descriptor for inner product weights gradient primitive.
|
static int |
dnnl.dnnl_inner_product_forward_desc_init(dnnl_inner_product_desc_t ip_desc,
int prop_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc)
\} dnnl_api_layer_normalization
|
static int |
dnnl.dnnl_layer_normalization_backward_desc_init(dnnl_layer_normalization_desc_t lnrm_desc,
int prop_kind,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
dnnl_memory_desc_t stat_desc,
float epsilon,
int flags)
Initializes a descriptor for a layer normalization backward propagation
primitive.
|
static int |
dnnl.dnnl_layer_normalization_forward_desc_init(dnnl_layer_normalization_desc_t lnrm_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
dnnl_memory_desc_t stat_desc,
float epsilon,
int flags)
\} dnnl_api_batch_normalization
|
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_logsoftmax_backward_desc_init(dnnl_softmax_desc_t logsoftmax_desc,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
int logsoftmax_axis)
Initializes a descriptor for logsoftmax backward propagation primitive.
|
static int |
dnnl.dnnl_logsoftmax_forward_desc_init(dnnl_softmax_desc_t logsoftmax_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
int logsoftmax_axis)
\} dnnl_api_softmax
|
static int |
dnnl.dnnl_lrn_backward_desc_init(dnnl_lrn_desc_t lrn_desc,
int alg_kind,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
long local_size,
float alpha,
float beta,
float k)
Initializes a descriptor for LRN backward propagation primitive.
|
static int |
dnnl.dnnl_lrn_forward_desc_init(dnnl_lrn_desc_t lrn_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t data_desc,
long local_size,
float alpha,
float beta,
float k)
\} dnnl_api_pooling
|
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_matmul_desc_init(dnnl_matmul_desc_t matmul_desc,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t weights_desc,
dnnl_memory_desc_t bias_desc,
dnnl_memory_desc_t dst_desc)
\} dnnl_api_rnn
|
static int |
dnnl.dnnl_memory_create(dnnl_memory memory,
dnnl_memory_desc_t memory_desc,
dnnl_engine engine,
Pointer handle)
Creates a memory object.
|
static int |
dnnl.dnnl_memory_create(PointerPointer memory,
dnnl_memory_desc_t memory_desc,
dnnl_engine engine,
Pointer handle) |
static int |
dnnl.dnnl_memory_desc_equal(dnnl_memory_desc_t lhs,
dnnl_memory_desc_t rhs)
Compares two memory descriptors.
|
static long |
dnnl.dnnl_memory_desc_get_size(dnnl_memory_desc_t memory_desc)
Returns the size of a memory descriptor.
|
static int |
dnnl.dnnl_memory_desc_init_by_strides(dnnl_memory_desc_t memory_desc,
int ndims,
long[] dims,
int data_type,
long[] strides) |
static int |
dnnl.dnnl_memory_desc_init_by_strides(dnnl_memory_desc_t memory_desc,
int ndims,
LongBuffer dims,
int data_type,
LongBuffer strides) |
static int |
dnnl.dnnl_memory_desc_init_by_strides(dnnl_memory_desc_t memory_desc,
int ndims,
LongPointer dims,
int data_type,
LongPointer strides)
\} dnnl_api_attributes
|
static int |
dnnl.dnnl_memory_desc_init_by_tag(dnnl_memory_desc_t memory_desc,
int ndims,
long[] dims,
int data_type,
int tag) |
static int |
dnnl.dnnl_memory_desc_init_by_tag(dnnl_memory_desc_t memory_desc,
int ndims,
LongBuffer dims,
int data_type,
int tag) |
static int |
dnnl.dnnl_memory_desc_init_by_tag(dnnl_memory_desc_t memory_desc,
int ndims,
LongPointer dims,
int data_type,
int tag)
Initializes a memory descriptor using dimensions and memory format tag.
|
static int |
dnnl.dnnl_memory_desc_init_submemory(dnnl_memory_desc_t memory_desc,
dnnl_memory_desc_t parent_memory_desc,
long[] dims,
long[] offsets) |
static int |
dnnl.dnnl_memory_desc_init_submemory(dnnl_memory_desc_t memory_desc,
dnnl_memory_desc_t parent_memory_desc,
LongBuffer dims,
LongBuffer offsets) |
static int |
dnnl.dnnl_memory_desc_init_submemory(dnnl_memory_desc_t memory_desc,
dnnl_memory_desc_t parent_memory_desc,
LongPointer dims,
LongPointer offsets) |
static int |
dnnl.dnnl_memory_desc_permute_axes(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
int[] permutation) |
static int |
dnnl.dnnl_memory_desc_permute_axes(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
IntBuffer permutation) |
static int |
dnnl.dnnl_memory_desc_permute_axes(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
IntPointer permutation)
Initializes a memory descriptor by permuting axes in an existing one.
|
static int |
dnnl.dnnl_memory_desc_reshape(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
int ndims,
long[] dims) |
static int |
dnnl.dnnl_memory_desc_reshape(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
int ndims,
LongBuffer dims) |
static int |
dnnl.dnnl_memory_desc_reshape(dnnl_memory_desc_t out_memory_desc,
dnnl_memory_desc_t in_memory_desc,
int ndims,
LongPointer dims)
Initializes a memory descriptor by reshaping an existing one.
|
static int |
dnnl.dnnl_memory_get_memory_desc(dnnl_memory memory,
dnnl_memory_desc_t memory_desc) |
static int |
dnnl.dnnl_pooling_backward_desc_init(dnnl_pooling_desc_t pool_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc,
long[] strides,
long[] kernel,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_pooling_backward_desc_init(dnnl_pooling_desc_t pool_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc,
LongBuffer strides,
LongBuffer kernel,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_pooling_backward_desc_init(dnnl_pooling_desc_t pool_desc,
int alg_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc,
LongPointer strides,
LongPointer kernel,
LongPointer padding_l,
LongPointer padding_r)
Initializes a descriptor for pooling backward propagation primitive.
|
static int |
dnnl.dnnl_pooling_forward_desc_init(dnnl_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc,
long[] strides,
long[] kernel,
long[] padding_l,
long[] padding_r) |
static int |
dnnl.dnnl_pooling_forward_desc_init(dnnl_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc,
LongBuffer strides,
LongBuffer kernel,
LongBuffer padding_l,
LongBuffer padding_r) |
static int |
dnnl.dnnl_pooling_forward_desc_init(dnnl_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc,
LongPointer strides,
LongPointer kernel,
LongPointer padding_l,
LongPointer padding_r)
\} dnnl_api_logsoftmax
|
static int |
dnnl.dnnl_reorder_primitive_desc_create(dnnl_primitive_desc reorder_primitive_desc,
dnnl_memory_desc_t src_desc,
dnnl_engine src_engine,
dnnl_memory_desc_t dst_desc,
dnnl_engine dst_engine,
dnnl_primitive_attr attr)
\} dnnl_api_memory
|
static int |
dnnl.dnnl_reorder_primitive_desc_create(PointerPointer reorder_primitive_desc,
dnnl_memory_desc_t src_desc,
dnnl_engine src_engine,
dnnl_memory_desc_t dst_desc,
dnnl_engine dst_engine,
dnnl_primitive_attr attr) |
static int |
dnnl.dnnl_resampling_backward_desc_init(dnnl_resampling_desc_t resampling_desc,
int alg_kind,
float[] factors,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc) |
static int |
dnnl.dnnl_resampling_backward_desc_init(dnnl_resampling_desc_t resampling_desc,
int alg_kind,
FloatBuffer factors,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc) |
static int |
dnnl.dnnl_resampling_backward_desc_init(dnnl_resampling_desc_t resampling_desc,
int alg_kind,
FloatPointer factors,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc)
Initializes a descriptor for resampling backward propagation primitive.
|
static int |
dnnl.dnnl_resampling_forward_desc_init(dnnl_resampling_desc_t resampling_desc,
int prop_kind,
int alg_kind,
float[] factors,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc) |
static int |
dnnl.dnnl_resampling_forward_desc_init(dnnl_resampling_desc_t resampling_desc,
int prop_kind,
int alg_kind,
FloatBuffer factors,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc) |
static int |
dnnl.dnnl_resampling_forward_desc_init(dnnl_resampling_desc_t resampling_desc,
int prop_kind,
int alg_kind,
FloatPointer factors,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc)
\} dnnl_api_matmul
|
static int |
dnnl.dnnl_shuffle_backward_desc_init(dnnl_shuffle_desc_t shuffle_desc,
dnnl_memory_desc_t diff_data_desc,
int axis,
long group_size)
Initializes a descriptor for shuffle backward propagation primitive.
|
static int |
dnnl.dnnl_shuffle_forward_desc_init(dnnl_shuffle_desc_t shuffle_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
int axis,
long group_size)
\} dnnl_api_deconvolution
|
static int |
dnnl.dnnl_softmax_backward_desc_init(dnnl_softmax_desc_t softmax_desc,
dnnl_memory_desc_t diff_data_desc,
dnnl_memory_desc_t data_desc,
int softmax_axis)
Initializes a descriptor for softmax backward propagation primitive.
|
static int |
dnnl.dnnl_softmax_forward_desc_init(dnnl_softmax_desc_t softmax_desc,
int prop_kind,
dnnl_memory_desc_t data_desc,
int softmax_axis)
\} dnnl_api_eltwise
|
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
float[] scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatBuffer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatPointer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine)
\} dnnl_api_concat
|
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
float[] scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatBuffer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatPointer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
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.