Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.bias_desc()
Bias memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.bias_desc()
Bias memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.bias_desc()
Bias memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_softmax_desc_t.data_desc()
Source and destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_eltwise_desc_t.data_desc()
Source and destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.data_desc()
Source and destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_lrn_desc_t.data_desc()
Source and destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_shuffle_desc_t.data_desc()
Source and destination memory descriptor,
and source and destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.data_scaleshift_desc()
Scale and shift data and gradient memory descriptors.
|
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.data_type(int setter) |
mkldnn_memory_desc_t |
memory.desc.data()
The underlying C API data structure.
|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.diff_bias_desc()
Bias gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_eltwise_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_lrn_desc_t.diff_data_desc()
Source and destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.diff_data_scaleshift_desc() |
mkldnn_memory_desc_t |
mkldnn_softmax_desc_t.diff_desc()
Source and Destination of gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_pooling_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.diff_dst_desc()
Destination gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_dst_iter_desc()
Destination gradient iteration memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_dst_layer_desc()
Destination gradient layer memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_pooling_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.diff_src_desc()
Source gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_src_iter_desc()
Source gradient iter memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_src_layer_desc()
Source gradient layer memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.diff_weights_desc()
Weights gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.diff_weights_desc()
Weights gradient memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_weights_iter_desc()
Weights gradient iter memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.diff_weights_layer_desc()
Weights gradient layer memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.dst_desc()
Destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_pooling_desc_t.dst_desc()
Destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.dst_desc()
Destination memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.dst_iter_desc()
Destination iter memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.dst_layer_desc()
Destination layer memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.format(int setter) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.getPointer(long i) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.layout_desc_blocking(mkldnn_blocking_desc_t setter) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.layout_desc_rnn_packed_desc(mkldnn_rnn_packed_desc_t setter) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.layout_desc_wino_desc(mkldnn_wino_desc_t setter) |
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.mean_desc()
Mean and variance data memory descriptors.
|
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.ndims(int setter) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.position(long position) |
mkldnn_memory_desc_t |
mkldnn_memory_desc_t.primitive_kind(int setter) |
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.src_desc()
Source memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_pooling_desc_t.src_desc()
Source memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.src_desc()
Source memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.src_iter_desc()
Source iteration memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.src_layer_desc()
Source layer memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_batch_normalization_desc_t.variance_desc() |
mkldnn_memory_desc_t |
mkldnn_inner_product_desc_t.weights_desc()
Weights memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_convolution_desc_t.weights_desc()
Weights memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.weights_iter_desc()
Weights iteration memory descriptor.
|
mkldnn_memory_desc_t |
mkldnn_rnn_desc_t.weights_layer_desc()
Weights layer memory descriptor.
|
Constructor and Description |
---|
desc(mkldnn_memory_desc_t adata)
Constructs a memory descriptor from a C API data structure.
|
Modifier and Type | Method and Description |
---|---|
static mkldnn_memory_desc_t |
mkldnn.mkldnn_primitive_desc_query_memory_d(mkldnn_primitive_desc primitive_desc)
Queries primitive descriptor for memory descriptor
|
Modifier and Type | Method and Description |
---|---|
static int |
mkldnn.mkldnn_batch_normalization_backward_desc_init(mkldnn_batch_normalization_desc_t bnrm_desc,
int prop_kind,
mkldnn_memory_desc_t diff_data_desc,
mkldnn_memory_desc_t data_desc,
float epsilon,
int flags)
Initializes a batch normalization descriptor \p bnrm_desc for backward
propagation with respect to data and scale-shift parameters using memory
descriptors \p data_desc and \p diff_data_desc, normalization parameter
\p epsilon, and \p flags set using bit flags of type
mkldnn_batch_normalization_desc_t.
|
static int |
mkldnn.mkldnn_batch_normalization_forward_desc_init(mkldnn_batch_normalization_desc_t bnrm_desc,
int prop_kind,
mkldnn_memory_desc_t data_desc,
float epsilon,
int flags)
Initializes a batch normalization descriptor \p bnrm_desc for forward
propagation using \p prop_kind (possible values are
#mkldnn_forward_training and #mkldnn_forward_inference), memory descriptor
\p data_desc, normalization parameter \p epsilon, and \p flags set using bit
flags of type mkldnn_batch_normalization_desc_t.
|
static int |
mkldnn.mkldnn_concat_primitive_desc_create(mkldnn_primitive_desc concat_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
int concat_dimension,
mkldnn_primitive_desc input_pds)
Creates out-of-place \p concat_primitive_desc for concatenation of \p n
inputs by \p concat_dimension with resulting \p output_desc memory
descriptor.
|
static int |
mkldnn.mkldnn_concat_primitive_desc_create(PointerPointer concat_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
int concat_dimension,
PointerPointer input_pds) |
static int |
mkldnn.mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a convolution descriptor \p conv_desc for backward propagation
with respect to data using \p alg_kind, memory descriptors, \p strides, \p
padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a convolution descriptor \p conv_desc for backward propagation
with respect to weights using \p alg_kind, memory descriptors, \p strides,
\p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a convolution descriptor \p conv_desc for forward propagation
using \p prop_kind (possible values are #mkldnn_forward_training and
#mkldnn_forward_inference), \p alg_kind, memory descriptors, \p strides, \p
padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a deconvolution descriptor \p conv_desc for backward propagation
with respect to data using \p alg_kind, memory descriptors, \p strides, \p
padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a deconvolution descriptor \p conv_desc for backward propagation
with respect to weights using \p alg_kind, memory descriptors, \p strides,
\p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
int[] strides,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntBuffer strides,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntPointer strides,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a deconvolution descriptor \p deconv_desc for forward
propagation using \p prop_kind (possible values are #mkldnn_forward_training
and #mkldnn_forward_inference), \p alg_kind, memory descriptors, \p strides,
\p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a dilated convolution descriptor \p conv_desc for backward
propagation with respect to data using \p alg_kind, memory descriptors, \p
strides, \p dilates \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a convolution descriptor \p conv_desc for backward propagation
with respect to weights using \p alg_kind, memory descriptors, \p strides,
\p dilates \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a dilated convolution descriptor \p conv_desc for forward
propagation using \p prop_kind (possible values are #mkldnn_forward_training
and #mkldnn_forward_inference), \p alg_kind, memory descriptors, \p strides,
\p dilates, \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a dilated deconvolution descriptor \p conv_desc for backward
propagation with respect to data using \p alg_kind, memory descriptors, \p
strides, \p dilates, \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a dilated deconvolution descriptor \p conv_desc for backward
propagation with respect to weights using \p alg_kind, memory descriptors,
\p strides, \p dilates, \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_dilated_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
int[] strides,
int[] dilates,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntBuffer strides,
IntBuffer dilates,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_dilated_deconvolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc,
IntPointer strides,
IntPointer dilates,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a dilated deconvolution descriptor \p deconv_desc for forward
propagation using \p prop_kind (possible values are #mkldnn_forward_training
and #mkldnn_forward_inference), \p alg_kind, memory descriptors, \p strides,
\p dilates, \p padding_l, \p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_eltwise_backward_desc_init(mkldnn_eltwise_desc_t eltwise_desc,
int alg_kind,
mkldnn_memory_desc_t diff_data_desc,
mkldnn_memory_desc_t data_desc,
float alpha,
float beta)
Initializes an \p eltwise_desc for backward propagation using \p alg_kind
algorithm memory descriptors \p diff_data_desc and \p data_desc, and the
\p alpha and \p beta parameters.
|
static int |
mkldnn.mkldnn_eltwise_forward_desc_init(mkldnn_eltwise_desc_t eltwise_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t data_desc,
float alpha,
float beta)
Initializes an \p eltwise_desc for forward propagation using \p prop_kind
(possible values are #mkldnn_forward_training and #mkldnn_forward_inference),
\p alg_kind algorithm, memory descriptor \p data_desc, \p alpha, and
\p beta parameters.
|
static int |
mkldnn.mkldnn_inner_product_backward_data_desc_init(mkldnn_inner_product_desc_t ip_desc,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t diff_dst_desc)
Initializes an inner product descriptor \p ip_desc for backward propagation
with respect to data using memory descriptors.
|
static int |
mkldnn.mkldnn_inner_product_backward_weights_desc_init(mkldnn_inner_product_desc_t ip_desc,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t diff_weights_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_desc)
Initializes an inner product descriptor \p ip_desc for backward propagation
with respect to weights using memory descriptors.
|
static int |
mkldnn.mkldnn_inner_product_forward_desc_init(mkldnn_inner_product_desc_t ip_desc,
int prop_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t weights_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_desc)
Initializes an inner product descriptor \p ip_desc for forward propagation
using \p prop_kind (possible values are #mkldnn_forward_training and
#mkldnn_forward_inference) and memory descriptors.
|
static int |
mkldnn.mkldnn_lrn_backward_desc_init(mkldnn_lrn_desc_t lrn_desc,
int alg_kind,
mkldnn_memory_desc_t diff_data_desc,
mkldnn_memory_desc_t data_desc,
int local_size,
float alpha,
float beta,
float k)
Initializes an \p lrn_desc for backward propagation using \p alg_kind,
memory descriptors \p data_desc and \p diff_data_desc, and regularization
parameters \p local_size, \p alpha, \p beta, and \p k.
|
static int |
mkldnn.mkldnn_lrn_forward_desc_init(mkldnn_lrn_desc_t lrn_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t data_desc,
int local_size,
float alpha,
float beta,
float k)
Initializes an \p lrn_desc for forward propagation using \p prop_kind
(possible values are #mkldnn_forward_training and #mkldnn_forward_inference),
\p alg_kind, memory descriptor \p data_desc, and regularization
parameters \p local_size, \p alpha, \p beta, and \p k.
|
static int |
mkldnn.mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc,
int ndims,
int[] dims,
int data_type,
int format) |
static int |
mkldnn.mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc,
int ndims,
IntBuffer dims,
int data_type,
int format) |
static int |
mkldnn.mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc,
int ndims,
IntPointer dims,
int data_type,
int format)
Initializes a \p memory_desc memory descriptor using \p ndims, \p dims, \p
data_type, and data \p format.
|
static int |
mkldnn.mkldnn_memory_primitive_desc_create(mkldnn_primitive_desc memory_primitive_desc,
mkldnn_memory_desc_t memory_desc,
mkldnn_engine engine)
Creates a \p memory_primitive_desc memory primitive descriptor using \p
memory_desc and \p engine.
|
static int |
mkldnn.mkldnn_memory_primitive_desc_create(PointerPointer memory_primitive_desc,
mkldnn_memory_desc_t memory_desc,
mkldnn_engine engine) |
static int |
mkldnn.mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t diff_dst_desc,
int[] strides,
int[] kernel,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntBuffer strides,
IntBuffer kernel,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc,
int alg_kind,
mkldnn_memory_desc_t diff_src_desc,
mkldnn_memory_desc_t diff_dst_desc,
IntPointer strides,
IntPointer kernel,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a pooling descriptor \p pool_desc for backward propagation
using \p alg_kind, memory descriptors, and pooling parameters in the spatial
domain: \p strides, \p kernel sizes, \p padding_l, \p padding_r, and \p
padding_kind.
|
static int |
mkldnn.mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t dst_desc,
int[] strides,
int[] kernel,
int[] padding_l,
int[] padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t dst_desc,
IntBuffer strides,
IntBuffer kernel,
IntBuffer padding_l,
IntBuffer padding_r,
int padding_kind) |
static int |
mkldnn.mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc,
int prop_kind,
int alg_kind,
mkldnn_memory_desc_t src_desc,
mkldnn_memory_desc_t dst_desc,
IntPointer strides,
IntPointer kernel,
IntPointer padding_l,
IntPointer padding_r,
int padding_kind)
Initializes a pooling descriptor \p pool_desc for forward propagation using
\p prop_kind (possible values are #mkldnn_forward_training and
#mkldnn_forward_inference), \p alg_kind, memory descriptors, and pooling
parameters in the spatial domain: \p strides, \p kernel sizes, \p padding_l,
\p padding_r, and \p padding_kind.
|
static int |
mkldnn.mkldnn_rnn_backward_desc_init(mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn_memory_desc_t src_layer_desc,
mkldnn_memory_desc_t src_iter_desc,
mkldnn_memory_desc_t weights_layer_desc,
mkldnn_memory_desc_t weights_iter_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_layer_desc,
mkldnn_memory_desc_t dst_iter_desc,
mkldnn_memory_desc_t diff_src_layer_desc,
mkldnn_memory_desc_t diff_src_iter_desc,
mkldnn_memory_desc_t diff_weights_layer_desc,
mkldnn_memory_desc_t diff_weights_iter_desc,
mkldnn_memory_desc_t diff_bias_desc,
mkldnn_memory_desc_t diff_dst_layer,
mkldnn_memory_desc_t diff_dst_iter_desc)
Initializes a rnn descriptor \p rnn_desc for backward propagation
using \p prop_kind, \p rnn_cell_desc, \p direction, and memory descriptors.
|
static int |
mkldnn.mkldnn_rnn_forward_desc_init(mkldnn_rnn_desc_t rnn_desc,
int prop_kind,
mkldnn_rnn_cell_desc_t rnn_cell_desc,
int direction,
mkldnn_memory_desc_t src_layer_desc,
mkldnn_memory_desc_t src_iter_desc,
mkldnn_memory_desc_t weights_layer_desc,
mkldnn_memory_desc_t weights_iter_desc,
mkldnn_memory_desc_t bias_desc,
mkldnn_memory_desc_t dst_layer_desc,
mkldnn_memory_desc_t dst_iter_desc)
Initializes a rnn descriptor \p rnn_desc for forward propagation
using \p prop_kind, \p rnn_cell_desc, \p direction, and memory descriptors.
|
static int |
mkldnn.mkldnn_shuffle_backward_desc_init(mkldnn_shuffle_desc_t shuffle_desc,
mkldnn_memory_desc_t diff_data_desc,
int axis,
int group_size)
Initializes a \p shuffle_desc for backward propagation using memory
descriptor \p diff_data_desc, \p axis, and \p group_size.
|
static int |
mkldnn.mkldnn_shuffle_forward_desc_init(mkldnn_shuffle_desc_t shuffle_desc,
int prop_kind,
mkldnn_memory_desc_t data_desc,
int axis,
int group_size)
Initializes a \p shuffle_desc for forward propagation using \p prop_kind,
memory descriptor \p data_desc, \p axis, and \p group_size.
|
static int |
mkldnn.mkldnn_softmax_backward_desc_init(mkldnn_softmax_desc_t softmax_desc,
mkldnn_memory_desc_t diff_desc,
mkldnn_memory_desc_t data_desc,
int softmax_axis)
Initializes a \p softmax_desc for backward propagation using memory
descriptors \p diff_desc and \p data_desc.
|
static int |
mkldnn.mkldnn_softmax_forward_desc_init(mkldnn_softmax_desc_t softmax_desc,
int prop_kind,
mkldnn_memory_desc_t data_desc,
int softmax_axis)
Initializes a \p softmax_desc for forward propagation using \p prop_kind
(possible values are #mkldnn_forward_training and #mkldnn_forward_inference)
and memory descriptor \p data_desc.
|
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
float[] scales,
mkldnn_primitive_desc input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatBuffer scales,
mkldnn_primitive_desc input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatPointer scales,
mkldnn_primitive_desc input_pds)
Creates out-of-place \p sum_primitive_desc for sum of \p n
inputs multiplied by scale with resulting \p output_desc memory
descriptor.
|
static int |
mkldnn.mkldnn_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
float[] scales,
PointerPointer input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatBuffer scales,
PointerPointer input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatPointer scales,
PointerPointer input_pds) |
Copyright © 2022. All rights reserved.