public class mkldnn extends mkldnn
Modifier and Type | Field and Description |
---|---|
static int |
algorithm_undef
enum mkldnn::algorithm
|
static int |
backward
enum mkldnn::prop_kind
|
static int |
backward_bias
enum mkldnn::prop_kind
|
static int |
backward_data
enum mkldnn::prop_kind
|
static int |
backward_weights
enum mkldnn::prop_kind
|
static int |
batch_normalization_d
enum mkldnn::query
|
static int |
bidirectional_concat
enum mkldnn::rnn_direction
|
static int |
bidirectional_sum
enum mkldnn::rnn_direction
|
static int |
convolution_auto
enum mkldnn::algorithm
|
static int |
convolution_d
enum mkldnn::query
|
static int |
convolution_direct
enum mkldnn::algorithm
|
static int |
convolution_winograd
enum mkldnn::algorithm
|
static int |
deconvolution_d
enum mkldnn::query
|
static int |
deconvolution_direct
enum mkldnn::algorithm
|
static int |
deconvolution_winograd
enum mkldnn::algorithm
|
static int |
diff_dst_pd
enum mkldnn::query
|
static int |
diff_src_pd
enum mkldnn::query
|
static int |
diff_weights_pd
enum mkldnn::query
|
static int |
dst_pd
enum mkldnn::query
|
static int |
eengine
enum mkldnn::query
|
static int |
eltwise_abs
enum mkldnn::algorithm
|
static int |
eltwise_bounded_relu
enum mkldnn::algorithm
|
static int |
eltwise_d
enum mkldnn::query
|
static int |
eltwise_elu
enum mkldnn::algorithm
|
static int |
eltwise_exp
enum mkldnn::algorithm
|
static int |
eltwise_gelu
enum mkldnn::algorithm
|
static int |
eltwise_linear
enum mkldnn::algorithm
|
static int |
eltwise_logistic
enum mkldnn::algorithm
|
static int |
eltwise_relu
enum mkldnn::algorithm
|
static int |
eltwise_soft_relu
enum mkldnn::algorithm
|
static int |
eltwise_sqrt
enum mkldnn::algorithm
|
static int |
eltwise_square
enum mkldnn::algorithm
|
static int |
eltwise_tanh
enum mkldnn::algorithm
|
static int |
forward
enum mkldnn::prop_kind
|
static int |
forward_inference
enum mkldnn::prop_kind
|
static int |
forward_scoring
enum mkldnn::prop_kind
|
static int |
forward_training
enum mkldnn::prop_kind
|
static int |
fuse_bn_relu
enum mkldnn::batch_normalization_flag
|
static int |
gru_linear_before_reset
enum mkldnn::algorithm
|
static int |
impl_info_str
enum mkldnn::query
|
static int |
inner_product_d
enum mkldnn::query
|
static int |
input_pd
enum mkldnn::query
|
static int |
lrn_across_channels
enum mkldnn::algorithm
|
static int |
lrn_d
enum mkldnn::query
|
static int |
lrn_within_channel
enum mkldnn::algorithm
|
static int |
memory_consumption_s64
enum mkldnn::query
|
static int |
memory_d
enum mkldnn::query
|
static int |
mkldnn_alg_kind_undef
enum mkldnn_alg_kind_t
|
static int |
mkldnn_any
enum mkldnn_memory_format_t
|
static int |
mkldnn_any_engine
enum mkldnn_engine_kind_t
|
static int |
mkldnn_any_stream
enum mkldnn_stream_kind_t
|
static int |
mkldnn_backward
enum mkldnn_prop_kind_t
|
static int |
mkldnn_backward_bias
enum mkldnn_prop_kind_t
|
static int |
mkldnn_backward_data
enum mkldnn_prop_kind_t
|
static int |
mkldnn_backward_weights
enum mkldnn_prop_kind_t
|
static int |
mkldnn_batch_normalization
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_bf16
enum mkldnn_data_type_t
|
static int |
mkldnn_bidirectional_concat
enum mkldnn_rnn_direction_t
|
static int |
mkldnn_bidirectional_sum
enum mkldnn_rnn_direction_t
|
static int |
mkldnn_blocked
enum mkldnn_memory_format_t
|
static int |
mkldnn_chwn
enum mkldnn_memory_format_t
|
static int |
mkldnn_concat
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_concat_inplace
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_convolution
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_convolution_auto
enum mkldnn_alg_kind_t
|
static int |
mkldnn_convolution_direct
enum mkldnn_alg_kind_t
|
static int |
mkldnn_convolution_winograd
enum mkldnn_alg_kind_t
|
static int |
mkldnn_cpu
enum mkldnn_engine_kind_t
|
static int |
mkldnn_data_type_undef
enum mkldnn_data_type_t
|
static int |
mkldnn_deconvolution
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_deconvolution_direct
enum mkldnn_alg_kind_t
|
static int |
mkldnn_deconvolution_winograd
enum mkldnn_alg_kind_t
|
static int |
mkldnn_dhwio
enum mkldnn_memory_format_t
|
static int |
mkldnn_eager
enum mkldnn_stream_kind_t
|
static int |
mkldnn_eager_nostore
enum mkldnn_stream_kind_t
|
static int |
mkldnn_eltwise
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_eltwise_abs
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_bounded_relu
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_elu
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_exp
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_gelu
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_linear
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_logistic
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_relu
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_soft_relu
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_sqrt
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_square
enum mkldnn_alg_kind_t
|
static int |
mkldnn_eltwise_tanh
enum mkldnn_alg_kind_t
|
static int |
mkldnn_f32
enum mkldnn_data_type_t
|
static int |
mkldnn_format_last
enum mkldnn_memory_format_t
|
static int |
mkldnn_format_undef
enum mkldnn_memory_format_t
|
static int |
mkldnn_forward
enum mkldnn_prop_kind_t
|
static int |
mkldnn_forward_inference
enum mkldnn_prop_kind_t
|
static int |
mkldnn_forward_scoring
enum mkldnn_prop_kind_t
|
static int |
mkldnn_forward_training
enum mkldnn_prop_kind_t
|
static int |
mkldnn_fuse_bn_relu
enum mkldnn_batch_normalization_flag_t
|
static int |
mkldnn_gIOdhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_giohw
enum mkldnn_memory_format_t
|
static int |
mkldnn_gIOhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gIOhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gIOhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gIOw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gIOw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOdhwi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOdhwi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOdhwi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOhIw16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOhwi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOhwi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOhwi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_goidhw
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOidhw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOidhw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIdhw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_goihw
enum mkldnn_memory_format_t
|
static int |
mkldnn_Goihw16g
enum mkldnn_memory_format_t
|
static int |
mkldnn_Goihw16g_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOihw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw2i8o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw2i8o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw4i16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw4i16o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOihw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw4o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw4o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_Goihw8g
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOihw8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIhw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_goiw
enum mkldnn_memory_format_t
|
static int |
mkldnn_Goiw16g
enum mkldnn_memory_format_t
|
static int |
mkldnn_Goiw16g_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOiw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw4i16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw4i16o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOiw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOIw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOwi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOwi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gOwi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_gru_linear_before_reset
enum mkldnn_alg_kind_t
|
static int |
mkldnn_hwigo
enum mkldnn_memory_format_t
|
static int |
mkldnn_hwigo_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_hwio
enum mkldnn_memory_format_t
|
static int |
mkldnn_hwio_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_ihwo
enum mkldnn_memory_format_t
|
static int |
mkldnn_inner_product
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_invalid_arguments
enum mkldnn_status_t
|
static int |
mkldnn_io
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOdhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_iohw
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_IOw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_iterator_ends
enum mkldnn_status_t
|
static int |
mkldnn_lazy
enum mkldnn_stream_kind_t
|
static int |
mkldnn_ldgo
enum mkldnn_memory_format_t
|
static int |
mkldnn_ldgoi
enum mkldnn_memory_format_t
|
static int |
mkldnn_ldgoi_p
enum mkldnn_rnn_packed_memory_format_t
|
static int |
mkldnn_ldigo
enum mkldnn_memory_format_t
|
static int |
mkldnn_ldigo_p
enum mkldnn_rnn_packed_memory_format_t
|
static int |
mkldnn_ldsnc
enum mkldnn_memory_format_t
|
static int |
mkldnn_lrn
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_lrn_across_channels
enum mkldnn_alg_kind_t
|
static int |
mkldnn_lrn_within_channel
enum mkldnn_alg_kind_t
|
static int |
mkldnn_memory
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_nc
enum mkldnn_memory_format_t
|
static int |
mkldnn_ncdhw
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCdhw16c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCdhw4c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCdhw8c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nchw
enum mkldnn_memory_format_t
|
static int |
mkldnn_nChw16c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nChw4c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nChw8c
enum mkldnn_memory_format_t
|
static int |
mkldnn_ncw
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCw16c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCw4c
enum mkldnn_memory_format_t
|
static int |
mkldnn_nCw8c
enum mkldnn_memory_format_t
|
static int |
mkldnn_ndhwc
enum mkldnn_memory_format_t
|
static int |
mkldnn_nhwc
enum mkldnn_memory_format_t
|
static int |
mkldnn_not_ready
enum mkldnn_status_t
|
static int |
mkldnn_not_required
enum mkldnn_status_t
|
static int |
mkldnn_ntc
enum mkldnn_memory_format_t
|
static int |
mkldnn_nwc
enum mkldnn_memory_format_t
|
static int |
mkldnn_odhwi
enum mkldnn_memory_format_t
|
static int |
mkldnn_Odhwi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Odhwi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Odhwi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OhIw16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_ohwi
enum mkldnn_memory_format_t
|
static int |
mkldnn_Ohwi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Ohwi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Ohwi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_oi
enum mkldnn_memory_format_t
|
static int |
mkldnn_oidhw
enum mkldnn_memory_format_t
|
static int |
mkldnn_oIdhw16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oidhw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oidhw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_oIdhw8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIdhw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_oihw
enum mkldnn_memory_format_t
|
static int |
mkldnn_oIhw16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oihw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw4i16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw4i16o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oihw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_oIhw8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oihw8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIhw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_oiw
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw16i16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oiw16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw16o16i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw4i16o4i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw4i16o4i_s8s8
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw4i4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Oiw4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw8i16o2i
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw8i8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw8o16i2o
enum mkldnn_memory_format_t
|
static int |
mkldnn_OIw8o8i
enum mkldnn_memory_format_t
|
static int |
mkldnn_out_of_memory
enum mkldnn_status_t
|
static int |
mkldnn_owi
enum mkldnn_memory_format_t
|
static int |
mkldnn_Owi16o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Owi4o
enum mkldnn_memory_format_t
|
static int |
mkldnn_Owi8o
enum mkldnn_memory_format_t
|
static int |
mkldnn_packed_format_undef
enum mkldnn_rnn_packed_memory_format_t
|
static int |
mkldnn_padding_zero
enum mkldnn_padding_kind_t
|
static int |
mkldnn_pooling
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_pooling_avg
enum mkldnn_alg_kind_t
|
static int |
mkldnn_pooling_avg_exclude_padding
enum mkldnn_alg_kind_t
|
static int |
mkldnn_pooling_avg_include_padding
enum mkldnn_alg_kind_t
|
static int |
mkldnn_pooling_max
enum mkldnn_alg_kind_t
|
static int |
mkldnn_prop_kind_undef
enum mkldnn_prop_kind_t
|
static int |
mkldnn_query_batch_normalization_d
enum mkldnn_query_t
|
static int |
mkldnn_query_convolution_d
enum mkldnn_query_t
|
static int |
mkldnn_query_deconvolution_d
enum mkldnn_query_t
|
static int |
mkldnn_query_diff_dst_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_diff_src_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_diff_weights_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_dst_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_eltwise_d
enum mkldnn_query_t
|
static int |
mkldnn_query_engine
enum mkldnn_query_t
|
static int |
mkldnn_query_impl_info_str
enum mkldnn_query_t
|
static int |
mkldnn_query_inner_product_d
enum mkldnn_query_t
|
static int |
mkldnn_query_input_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_lrn_d
enum mkldnn_query_t
|
static int |
mkldnn_query_memory_consumption_s64
enum mkldnn_query_t
|
static int |
mkldnn_query_memory_d
enum mkldnn_query_t
|
static int |
mkldnn_query_num_of_inputs_s32
enum mkldnn_query_t
|
static int |
mkldnn_query_num_of_outputs_s32
enum mkldnn_query_t
|
static int |
mkldnn_query_op_d
enum mkldnn_query_t
|
static int |
mkldnn_query_output_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_pooling_d
enum mkldnn_query_t
|
static int |
mkldnn_query_primitive_kind
enum mkldnn_query_t
|
static int |
mkldnn_query_rnn_d
enum mkldnn_query_t
|
static int |
mkldnn_query_shuffle_d
enum mkldnn_query_t
|
static int |
mkldnn_query_softmax_d
enum mkldnn_query_t
|
static int |
mkldnn_query_some_d
enum mkldnn_query_t
|
static int |
mkldnn_query_some_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_src_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_time_estimate_f64
enum mkldnn_query_t
|
static int |
mkldnn_query_undef
enum mkldnn_query_t
|
static int |
mkldnn_query_weights_pd
enum mkldnn_query_t
|
static int |
mkldnn_query_workspace_pd
enum mkldnn_query_t
|
static int |
mkldnn_reorder
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_rnn
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_rnn_cell_with_clipping
enum mkldnn_rnn_cell_flags_t
|
static int |
mkldnn_rnn_cell_with_relu
enum mkldnn_rnn_cell_flags_t
|
static int |
MKLDNN_RNN_MAX_N_PARTS |
static int |
mkldnn_rnn_packed
enum mkldnn_memory_format_t
|
static int |
mkldnn_round_down
enum mkldnn_round_mode_t
|
static int |
mkldnn_round_nearest
enum mkldnn_round_mode_t
|
static int |
mkldnn_runtime_error
enum mkldnn_status_t
|
static int |
mkldnn_s16
enum mkldnn_data_type_t
|
static int |
mkldnn_s32
enum mkldnn_data_type_t
|
static int |
mkldnn_s8
enum mkldnn_data_type_t
|
static int |
mkldnn_shuffle
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_softmax
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_success
enum mkldnn_status_t
|
static int |
mkldnn_sum
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_tnc
enum mkldnn_memory_format_t
|
static int |
mkldnn_try_again
enum mkldnn_status_t
|
static int |
mkldnn_u8
enum mkldnn_data_type_t
|
static int |
mkldnn_undefined_primitive
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_unidirectional
enum mkldnn_rnn_direction_t
|
static int |
mkldnn_unidirectional_left2right
enum mkldnn_rnn_direction_t
|
static int |
mkldnn_unidirectional_right2left
enum mkldnn_rnn_direction_t
|
static int |
mkldnn_unimplemented
enum mkldnn_status_t
|
static int |
mkldnn_use_global_stats
enum mkldnn_batch_normalization_flag_t
|
static int |
mkldnn_use_scaleshift
enum mkldnn_batch_normalization_flag_t
|
static int |
mkldnn_vanilla_gru
enum mkldnn_alg_kind_t
|
static int |
mkldnn_vanilla_lstm
enum mkldnn_alg_kind_t
|
static int |
mkldnn_vanilla_rnn
enum mkldnn_alg_kind_t
|
static int |
mkldnn_view
enum mkldnn_primitive_kind_t
|
static int |
mkldnn_wino_fmt
enum mkldnn_memory_format_t
|
static int |
mkldnn_wino_undef
enum mkldnn_wino_memory_format_t
|
static int |
mkldnn_wino_wei_aaOBiOo
enum mkldnn_wino_memory_format_t
|
static int |
mkldnn_wino_wei_aaOio
enum mkldnn_wino_memory_format_t
|
static int |
mkldnn_wino_wei_aaOIoi
enum mkldnn_wino_memory_format_t
|
static int |
mkldnn_wino_wei_OBaaIBOIio
enum mkldnn_wino_memory_format_t
|
static int |
mkldnn_wio
enum mkldnn_memory_format_t
|
static int |
mkldnn_x
enum mkldnn_memory_format_t
|
static int |
num_of_inputs_s32
enum mkldnn::query
|
static int |
num_of_outputs_s32
enum mkldnn::query
|
static int |
op_d
enum mkldnn::query
|
static int |
output_pd
enum mkldnn::query
|
static int |
pooling_avg
enum mkldnn::algorithm
|
static int |
pooling_avg_exclude_padding
enum mkldnn::algorithm
|
static int |
pooling_avg_include_padding
enum mkldnn::algorithm
|
static int |
pooling_d
enum mkldnn::query
|
static int |
pooling_max
enum mkldnn::algorithm
|
static int |
primitive_kind
enum mkldnn::query
|
static int |
rnn_d
enum mkldnn::query
|
static int |
round_down
enum mkldnn::round_mode
|
static int |
round_nearest
enum mkldnn::round_mode
|
static int |
shuffle_d
enum mkldnn::query
|
static int |
softmax_d
enum mkldnn::query
|
static int |
src_pd
enum mkldnn::query
|
static int |
TENSOR_MAX_DIMS
Maximum number of dimensions a tensor can have.
|
static int |
time_estimate_f64
enum mkldnn::query
|
static int |
undef
enum mkldnn::query
|
static int |
unidirectional
enum mkldnn::rnn_direction
|
static int |
unidirectional_left2right
enum mkldnn::rnn_direction
|
static int |
unidirectional_right2left
enum mkldnn::rnn_direction
|
static int |
use_global_stats
enum mkldnn::batch_normalization_flag
|
static int |
use_scale_shift
enum mkldnn::batch_normalization_flag
|
static int |
vanilla_gru
enum mkldnn::algorithm
|
static int |
vanilla_lstm
enum mkldnn::algorithm
|
static int |
vanilla_rnn
enum mkldnn::algorithm
|
static int |
weights_pd
enum mkldnn::query
|
static int |
workspace_pd
enum mkldnn::query
|
static int |
zero
enum mkldnn::padding_kind
|
Constructor and Description |
---|
mkldnn() |
Modifier and Type | Method and Description |
---|---|
static void |
check_num_parameters(mkldnn_primitive_desc aprimitive_desc,
int n_inputs,
int n_outputs,
BytePointer prim_name) |
static void |
check_num_parameters(mkldnn_primitive_desc aprimitive_desc,
int n_inputs,
int n_outputs,
String prim_name) |
static int |
convert_to_c(int akind)
\addtogroup cpp_api C++ API
\{
|
static boolean |
equals(int a,
int b) |
static boolean |
is_null_memory(mkldnn_primitive aprimitive) |
static int |
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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_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_engine_create(mkldnn_engine engine,
int kind,
long index)
Creates an \p engine of particular \p kind and \p index.
|
static int |
mkldnn_engine_create(PointerPointer engine,
int kind,
long index) |
static int |
mkldnn_engine_destroy(mkldnn_engine engine)
Destroys an \p engine.
|
static long |
mkldnn_engine_get_count(int kind)
Returns the number of engines of a particular \p kind.
|
static int |
mkldnn_engine_get_kind(mkldnn_engine engine,
int[] kind) |
static int |
mkldnn_engine_get_kind(mkldnn_engine engine,
IntBuffer kind) |
static int |
mkldnn_engine_get_kind(mkldnn_engine engine,
IntPointer kind)
Returns the kind of an \p engine.
|
static int |
mkldnn_gemm_bf16bf16f32(BytePointer transa,
BytePointer transb,
int[] M,
int[] N,
int[] K,
float[] alpha,
short[] A,
int[] lda,
short[] B,
int[] ldb,
float[] beta,
float[] c,
int[] ldc) |
static int |
mkldnn_gemm_bf16bf16f32(BytePointer transa,
BytePointer transb,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ShortBuffer A,
IntBuffer lda,
ShortBuffer B,
IntBuffer ldb,
FloatBuffer beta,
FloatBuffer c,
IntBuffer ldc) |
static int |
mkldnn_gemm_bf16bf16f32(BytePointer transa,
BytePointer transb,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
ShortPointer A,
IntPointer lda,
ShortPointer B,
IntPointer ldb,
FloatPointer beta,
FloatPointer c,
IntPointer ldc)
gemm_bf16bf16f32 performs a matrix-matrix multiplication operation defined
as
C := alpha*op( A )*op( B ) + beta*C
where
- op( X ) is one of op( X ) = X or op( X ) = X**T,
- alpha and beta are scalars,
- A, B and C are matrices, with op( A ) an m by k matrix, op( B ) a k by n
matrix and C an m by n matrix.
|
static int |
mkldnn_gemm_bf16bf16f32(String transa,
String transb,
int[] M,
int[] N,
int[] K,
float[] alpha,
short[] A,
int[] lda,
short[] B,
int[] ldb,
float[] beta,
float[] c,
int[] ldc) |
static int |
mkldnn_gemm_bf16bf16f32(String transa,
String transb,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ShortBuffer A,
IntBuffer lda,
ShortBuffer B,
IntBuffer ldb,
FloatBuffer beta,
FloatBuffer c,
IntBuffer ldc) |
static int |
mkldnn_gemm_bf16bf16f32(String transa,
String transb,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
ShortPointer A,
IntPointer lda,
ShortPointer B,
IntPointer ldb,
FloatPointer beta,
FloatPointer c,
IntPointer ldc) |
static int |
mkldnn_gemm_s8s8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
int[] M,
int[] N,
int[] K,
float[] alpha,
byte[] A,
int[] lda,
byte[] ao,
byte[] B,
int[] ldb,
byte[] bo,
float[] beta,
int[] c,
int[] ldc,
int[] co) |
static int |
mkldnn_gemm_s8s8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ByteBuffer A,
IntBuffer lda,
ByteBuffer ao,
ByteBuffer B,
IntBuffer ldb,
ByteBuffer bo,
FloatBuffer beta,
IntBuffer c,
IntBuffer ldc,
IntBuffer co) |
static int |
mkldnn_gemm_s8s8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
BytePointer A,
IntPointer lda,
BytePointer ao,
BytePointer B,
IntPointer ldb,
BytePointer bo,
FloatPointer beta,
IntPointer c,
IntPointer ldc,
IntPointer co) |
static int |
mkldnn_gemm_s8s8s32(String transa,
String transb,
String offsetc,
int[] M,
int[] N,
int[] K,
float[] alpha,
byte[] A,
int[] lda,
byte[] ao,
byte[] B,
int[] ldb,
byte[] bo,
float[] beta,
int[] c,
int[] ldc,
int[] co) |
static int |
mkldnn_gemm_s8s8s32(String transa,
String transb,
String offsetc,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ByteBuffer A,
IntBuffer lda,
ByteBuffer ao,
ByteBuffer B,
IntBuffer ldb,
ByteBuffer bo,
FloatBuffer beta,
IntBuffer c,
IntBuffer ldc,
IntBuffer co) |
static int |
mkldnn_gemm_s8s8s32(String transa,
String transb,
String offsetc,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
BytePointer A,
IntPointer lda,
BytePointer ao,
BytePointer B,
IntPointer ldb,
BytePointer bo,
FloatPointer beta,
IntPointer c,
IntPointer ldc,
IntPointer co) |
static int |
mkldnn_gemm_s8u8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
int[] M,
int[] N,
int[] K,
float[] alpha,
byte[] A,
int[] lda,
byte[] ao,
byte[] B,
int[] ldb,
byte[] bo,
float[] beta,
int[] c,
int[] ldc,
int[] co) |
static int |
mkldnn_gemm_s8u8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ByteBuffer A,
IntBuffer lda,
ByteBuffer ao,
ByteBuffer B,
IntBuffer ldb,
ByteBuffer bo,
FloatBuffer beta,
IntBuffer c,
IntBuffer ldc,
IntBuffer co) |
static int |
mkldnn_gemm_s8u8s32(BytePointer transa,
BytePointer transb,
BytePointer offsetc,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
BytePointer A,
IntPointer lda,
BytePointer ao,
BytePointer B,
IntPointer ldb,
BytePointer bo,
FloatPointer beta,
IntPointer c,
IntPointer ldc,
IntPointer co)
gemm_s8u8s32 and gemm_s8s8s32 perform a matrix-matrix multiplication
operation and add the result to a scalar-matrix product.
|
static int |
mkldnn_gemm_s8u8s32(String transa,
String transb,
String offsetc,
int[] M,
int[] N,
int[] K,
float[] alpha,
byte[] A,
int[] lda,
byte[] ao,
byte[] B,
int[] ldb,
byte[] bo,
float[] beta,
int[] c,
int[] ldc,
int[] co) |
static int |
mkldnn_gemm_s8u8s32(String transa,
String transb,
String offsetc,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
ByteBuffer A,
IntBuffer lda,
ByteBuffer ao,
ByteBuffer B,
IntBuffer ldb,
ByteBuffer bo,
FloatBuffer beta,
IntBuffer c,
IntBuffer ldc,
IntBuffer co) |
static int |
mkldnn_gemm_s8u8s32(String transa,
String transb,
String offsetc,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
BytePointer A,
IntPointer lda,
BytePointer ao,
BytePointer B,
IntPointer ldb,
BytePointer bo,
FloatPointer beta,
IntPointer c,
IntPointer ldc,
IntPointer co) |
static int |
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_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_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_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_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_memory_desc_init(mkldnn_memory_desc_t memory_desc,
int ndims,
int[] dims,
int data_type,
int format) |
static int |
mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc,
int ndims,
IntBuffer dims,
int data_type,
int format) |
static int |
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_memory_get_data_handle(mkldnn_primitive memory,
Pointer handle) |
static int |
mkldnn_memory_get_data_handle(mkldnn_primitive memory,
PointerPointer handle)
For a \p memory primitive, returns the data \p handle.
|
static int |
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_memory_primitive_desc_create(PointerPointer memory_primitive_desc,
mkldnn_memory_desc_t memory_desc,
mkldnn_engine engine) |
static int |
mkldnn_memory_primitive_desc_equal(mkldnn_primitive_desc lhs,
mkldnn_primitive_desc rhs)
Compares two descriptors of memory primitives.
|
static long |
mkldnn_memory_primitive_desc_get_size(mkldnn_primitive_desc memory_primitive_desc)
Returns the size (in bytes) that is required for given \p
memory_primitive_desc
|
static int |
mkldnn_memory_set_data_handle(mkldnn_primitive memory,
Pointer handle)
For a \p memory primitive, sets the data \p handle.
|
static int |
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_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_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_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_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_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_post_ops_append_eltwise(mkldnn_post_ops post_ops,
float scale,
int alg,
float alpha,
float beta)
Appends eltwise post operation to the \p post_ops with given parameters
\p kind, \p alpha, and \p beta (@see mkldnn_eltwise_forward_desc_init and
mkldnn_eltwise_desc_t).
|
static int |
mkldnn_post_ops_append_sum(mkldnn_post_ops post_ops,
float scale)
Appends accumulation (sum) post operation to the \p post_ops.
|
static int |
mkldnn_post_ops_create(mkldnn_post_ops post_ops)
Creates an empty sequence of post operations \p post_ops.
|
static int |
mkldnn_post_ops_create(PointerPointer post_ops) |
static int |
mkldnn_post_ops_destroy(mkldnn_post_ops post_ops)
Deletes a \p post_ops sequence.
|
static int |
mkldnn_post_ops_get_kind(mkldnn_post_ops post_ops,
int index)
Returns the type of post operation with index \p index in given
\p post_ops.
|
static int |
mkldnn_post_ops_get_params_eltwise(mkldnn_post_ops post_ops,
int index,
float[] scale,
int[] alg,
float[] alpha,
float[] beta) |
static int |
mkldnn_post_ops_get_params_eltwise(mkldnn_post_ops post_ops,
int index,
FloatBuffer scale,
IntBuffer alg,
FloatBuffer alpha,
FloatBuffer beta) |
static int |
mkldnn_post_ops_get_params_eltwise(mkldnn_post_ops post_ops,
int index,
FloatPointer scale,
IntPointer alg,
FloatPointer alpha,
FloatPointer beta)
Gets the eltwise parameters of the post operation with index \p index in
the sequence of \p post_ops.
|
static int |
mkldnn_post_ops_get_params_sum(mkldnn_post_ops post_ops,
int index,
float[] scale) |
static int |
mkldnn_post_ops_get_params_sum(mkldnn_post_ops post_ops,
int index,
FloatBuffer scale) |
static int |
mkldnn_post_ops_get_params_sum(mkldnn_post_ops post_ops,
int index,
FloatPointer scale)
Gets the parameters of the accumulation (sum) post operation with index
\p index in the sequence of \p post_ops.
|
static int |
mkldnn_post_ops_len(mkldnn_post_ops post_ops)
Returns the \p length of post operations for given \p post_ops.
|
static mkldnn_primitive_at_t |
mkldnn_primitive_at(mkldnn_primitive primitive,
long output_index)
Creates an #mkldnn_primitive_at_t structure from a \p primitive and \p
output_index.
|
static int |
mkldnn_primitive_attr_clone(mkldnn_primitive_attr attr,
mkldnn_primitive_attr existing_attr)
Makes a copy of an \p existing_attr.
|
static int |
mkldnn_primitive_attr_clone(PointerPointer attr,
mkldnn_primitive_attr existing_attr) |
static int |
mkldnn_primitive_attr_create(mkldnn_primitive_attr attr)
Creates an empty (default) \p attr attribute.
|
static int |
mkldnn_primitive_attr_create(PointerPointer attr) |
static int |
mkldnn_primitive_attr_destroy(mkldnn_primitive_attr attr)
Deletes an \p attr.
|
static int |
mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
int[] round_mode) |
static int |
mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
IntBuffer round_mode) |
static int |
mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
IntPointer round_mode)
Returns integer output rounding mode \p round_mode for a given \p attr,
previously set by mkldnn_primitive_attr_set_int_output_round_mode.
|
static int |
mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
int[] count,
int[] mask,
float[] scales) |
static int |
mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntBuffer count,
IntBuffer mask,
FloatBuffer scales) |
static int |
mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntPointer count,
IntPointer mask,
FloatPointer scales) |
static int |
mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntPointer count,
IntPointer mask,
PointerPointer scales)
Returns \p count, correspondence scale \p mask, and a pointer to a constant
floating point array of output \p scales for given \p attr, previously set
by mkldnn_primitive_attr_set_output_scales.
|
static int |
mkldnn_primitive_attr_get_post_ops(mkldnn_primitive_attr attr,
mkldnn_post_ops post_ops)
Returns \p post_ops for given \p attr.
|
static int |
mkldnn_primitive_attr_get_post_ops(mkldnn_primitive_attr attr,
PointerPointer post_ops) |
static int |
mkldnn_primitive_attr_set_int_output_round_mode(mkldnn_primitive_attr attr,
int round_mode)
Sets output rounding mode \p round_mode for integer operations for a given
\p attr.
|
static int |
mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
float[] scales) |
static int |
mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
FloatBuffer scales) |
static int |
mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
FloatPointer scales)
Sets output \p scales for primitive operations.
|
static int |
mkldnn_primitive_attr_set_post_ops(mkldnn_primitive_attr attr,
mkldnn_post_ops post_ops)
Sets configured \p post_ops to an attribute \p attr for future use (when
primitive descriptor is being created).
|
static int |
mkldnn_primitive_attr_set_rnn_data_qparams(mkldnn_primitive_attr attr,
float scale,
float shift)
Sets quantization \p scale and \p shift for RNN data tensors.
|
static int |
mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
float[] weights_scales) |
static int |
mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
FloatBuffer weights_scales) |
static int |
mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
FloatPointer weights_scales)
Sets quantization scales \p weights_scales for RNN weights tensors.
|
static int |
mkldnn_primitive_create(mkldnn_primitive primitive,
mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_at_t inputs,
mkldnn_primitive outputs)
Creates a \p primitive using a \p primitive_desc descriptor and arrays of
\p inputs and \p outputs.
|
static int |
mkldnn_primitive_create(PointerPointer primitive,
mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_at_t inputs,
PointerPointer outputs) |
static int |
mkldnn_primitive_desc_clone(mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_desc existing_primitive_desc)
Makes a copy of a \p primitive_desc.
|
static int |
mkldnn_primitive_desc_clone(PointerPointer primitive_desc,
mkldnn_primitive_desc existing_primitive_desc) |
static int |
mkldnn_primitive_desc_create_v2(mkldnn_primitive_desc primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a \p primitive_desc using \p op_desc, \p attr, \p engine, and
optionally a hint primitive descriptor from forward propagation.
|
static int |
mkldnn_primitive_desc_create_v2(PointerPointer primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn_primitive_desc_create(mkldnn_primitive_desc primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a \p primitive_desc using \p op_desc, \p engine, and optionally a
hint primitive descriptor from forward propagation.
|
static int |
mkldnn_primitive_desc_create(PointerPointer primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn_primitive_desc_destroy(mkldnn_primitive_desc primitive_desc)
Deletes a \p primitive_desc.
|
static int |
mkldnn_primitive_desc_get_attr(mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_attr attr)
Returns a constant reference to the attribute of a \p primitive_desc.
|
static int |
mkldnn_primitive_desc_get_attr(mkldnn_primitive_desc primitive_desc,
PointerPointer attr) |
static int |
mkldnn_primitive_desc_iterator_create_v2(mkldnn_primitive_desc_iterator iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a primitive descriptor \p iterator for given \p op_desc, \p attr,
\p engine, and optionally a hint primitive descriptor from forward
propagation (required for backward propagation).
|
static int |
mkldnn_primitive_desc_iterator_create_v2(PointerPointer iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn_primitive_desc_iterator_create(mkldnn_primitive_desc_iterator iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a primitive descriptor \p iterator for given \p op_desc, \p engine,
and optionally a hint primitive descriptor from forward propagation
(required for backward propagation).
|
static int |
mkldnn_primitive_desc_iterator_create(PointerPointer iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn_primitive_desc_iterator_destroy(mkldnn_primitive_desc_iterator iterator)
Deletes a primitive descriptor \p iterator
|
static mkldnn_primitive_desc |
mkldnn_primitive_desc_iterator_fetch(mkldnn_primitive_desc_iterator iterator)
Fetches the current primitive descriptor.
|
static int |
mkldnn_primitive_desc_iterator_next(mkldnn_primitive_desc_iterator iterator)
Iterates over primitive descriptors.
|
static mkldnn_memory_desc_t |
mkldnn_primitive_desc_query_memory_d(mkldnn_primitive_desc primitive_desc)
Queries primitive descriptor for memory descriptor
|
static mkldnn_primitive_desc |
mkldnn_primitive_desc_query_pd(mkldnn_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for primitive descriptor
|
static int |
mkldnn_primitive_desc_query_s32(mkldnn_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for signed 32bit int
|
static int |
mkldnn_primitive_desc_query(mkldnn_primitive_desc primitive_desc,
int what,
int index,
Pointer result)
Queries primitive descriptor
One of the most typical use cases is to query a convolution primitive
descriptor created with source, weights, and destination formats equal
to #mkldnn_any about the corresponding memory primitive descriptors
(\p what equals #mkldnn_query_src_pd, #mkldnn_query_weights_pd, and
#mkldnn_query_dst_pd respectively) to be able to prepare memory and
create reorders if required.
|
static int |
mkldnn_primitive_destroy(mkldnn_primitive primitive)
Deletes a \p primitive.
|
static int |
mkldnn_primitive_get_input_at(mkldnn_primitive primitive,
long index,
mkldnn_primitive_at_t input)
For a \p primitive, returns \p input at the \p index position.
|
static int |
mkldnn_primitive_get_output(mkldnn_primitive primitive,
long index,
mkldnn_primitive output)
For a \p primitive, returns \p output at the \p index position.
|
static int |
mkldnn_primitive_get_output(mkldnn_primitive primitive,
long index,
PointerPointer output) |
static int |
mkldnn_primitive_get_primitive_desc(mkldnn_primitive primitive,
mkldnn_primitive_desc primitive_desc)
Retrieves a reference to the \p primitive_desc descriptor of given \p
primitive.
|
static int |
mkldnn_primitive_get_primitive_desc(mkldnn_primitive primitive,
PointerPointer primitive_desc) |
static int |
mkldnn_reorder_primitive_desc_create_v2(mkldnn_primitive_desc reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output,
mkldnn_primitive_attr attr)
Initializes a \p reorder_primitive_desc using an \p attr attribute and
descriptors of \p input and \p output memory primitives.
|
static int |
mkldnn_reorder_primitive_desc_create_v2(PointerPointer reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output,
mkldnn_primitive_attr attr) |
static int |
mkldnn_reorder_primitive_desc_create(mkldnn_primitive_desc reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output)
Initializes a \p reorder_primitive_desc using descriptors of \p input and
\p output memory primitives.
|
static int |
mkldnn_reorder_primitive_desc_create(PointerPointer reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output) |
static int |
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_rnn_cell_desc_init(mkldnn_rnn_cell_desc_t rnn_cell_desc,
int kind,
int f,
int flags,
float alpha,
float clipping)
Initializes a recurrent cell descriptor \p rnn_cell_desc
using \p rnn_cell_desc, \p kind (possible values are
#mkldnn_vanilla_rnn, #mkldnn_vanilla_lstm, #mkldnn_vanilla_gru, and
#mkldnn_gru_linear_before_reset),
\p f (possible values are #mkldnn_eltwise_relu and
#mkldnn_eltwise_tanh), \p flags, \p alpha, and \p clipping.
|
static int |
mkldnn_rnn_cell_get_gates_count(mkldnn_rnn_cell_desc_t rnn_cell_desc)
Returns the number of gates of a particular \p rnn_cell_desc.
|
static int |
mkldnn_rnn_cell_get_states_count(mkldnn_rnn_cell_desc_t rnn_cell_desc)
Returns the number of states of a particular \p rnn_cell_desc.
|
static int |
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_set_jit_dump(int dump)
Sets jit dump control.
|
static int |
mkldnn_set_verbose(int level)
Sets verbosity level (print information to stdout).
|
static int |
mkldnn_sgemm(BytePointer transa,
BytePointer transb,
int[] M,
int[] N,
int[] K,
float[] alpha,
float[] A,
int[] lda,
float[] B,
int[] ldb,
float[] beta,
float[] C,
int[] ldc) |
static int |
mkldnn_sgemm(BytePointer transa,
BytePointer transb,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
FloatBuffer A,
IntBuffer lda,
FloatBuffer B,
IntBuffer ldb,
FloatBuffer beta,
FloatBuffer C,
IntBuffer ldc) |
static int |
mkldnn_sgemm(BytePointer transa,
BytePointer transb,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
FloatPointer A,
IntPointer lda,
FloatPointer B,
IntPointer ldb,
FloatPointer beta,
FloatPointer C,
IntPointer ldc)
SGEMM performs a matrix-matrix multiplication operation defined as
C := alpha*op( A )*op( B ) + beta*C
where
- op( X ) is one of op( X ) = X or op( X ) = X**T,
- alpha and beta are scalars,
- A, B and C are matrices, with op( A ) an m by k matrix, op( B ) a k by n matrix
and C an m by n matrix.
|
static int |
mkldnn_sgemm(String transa,
String transb,
int[] M,
int[] N,
int[] K,
float[] alpha,
float[] A,
int[] lda,
float[] B,
int[] ldb,
float[] beta,
float[] C,
int[] ldc) |
static int |
mkldnn_sgemm(String transa,
String transb,
IntBuffer M,
IntBuffer N,
IntBuffer K,
FloatBuffer alpha,
FloatBuffer A,
IntBuffer lda,
FloatBuffer B,
IntBuffer ldb,
FloatBuffer beta,
FloatBuffer C,
IntBuffer ldc) |
static int |
mkldnn_sgemm(String transa,
String transb,
IntPointer M,
IntPointer N,
IntPointer K,
FloatPointer alpha,
FloatPointer A,
IntPointer lda,
FloatPointer B,
IntPointer ldb,
FloatPointer beta,
FloatPointer C,
IntPointer ldc) |
static int |
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_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_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_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_stream_create(mkldnn_stream stream,
int stream_kind)
Creates an execution \p stream of \p stream_kind.
|
static int |
mkldnn_stream_create(PointerPointer stream,
int stream_kind) |
static int |
mkldnn_stream_destroy(mkldnn_stream stream)
Destroys an execution \p stream.
|
static int |
mkldnn_stream_rerun(mkldnn_stream stream,
mkldnn_primitive error_primitive)
Reruns all the primitives within the \p stream.
|
static int |
mkldnn_stream_rerun(mkldnn_stream stream,
PointerPointer error_primitive) |
static int |
mkldnn_stream_submit(mkldnn_stream stream,
long n,
mkldnn_primitive primitives,
mkldnn_primitive error_primitive)
Submits \p primitives to an execution \p stream.
|
static int |
mkldnn_stream_submit(mkldnn_stream stream,
long n,
PointerPointer primitives,
PointerPointer error_primitive) |
static int |
mkldnn_stream_wait(mkldnn_stream stream,
int block,
mkldnn_primitive error_primitive)
Waits for all primitives in the execution \p stream to finish.
|
static int |
mkldnn_stream_wait(mkldnn_stream stream,
int block,
PointerPointer error_primitive) |
static int |
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_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_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_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
float[] scales,
PointerPointer input_pds) |
static int |
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_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatPointer scales,
PointerPointer input_pds) |
static mkldnn_version_t |
mkldnn_version()
Gets library version information.
|
static int |
mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
int[] dims,
int[] offsets) |
static int |
mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntBuffer dims,
IntBuffer offsets) |
static int |
mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntPointer dims,
IntPointer offsets)
Creates a \p view_primitive_desc for a given \p memory_primitive_desc, with
\p dims sizes and \p offsets offsets.
|
static int |
mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
int[] dims,
int[] offsets) |
static int |
mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntBuffer dims,
IntBuffer offsets) |
static int |
mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntPointer dims,
IntPointer offsets) |
static boolean |
notEquals(int a,
int b) |
static memory |
null_memory(engine eng) |
static memory.desc |
zero_md() |
public static final int mkldnn_success
public static final int mkldnn_out_of_memory
public static final int mkldnn_try_again
public static final int mkldnn_invalid_arguments
public static final int mkldnn_not_ready
public static final int mkldnn_unimplemented
public static final int mkldnn_iterator_ends
public static final int mkldnn_runtime_error
public static final int mkldnn_not_required
public static final int mkldnn_data_type_undef
public static final int mkldnn_f32
public static final int mkldnn_s32
public static final int mkldnn_s16
public static final int mkldnn_s8
public static final int mkldnn_u8
public static final int mkldnn_bf16
public static final int mkldnn_round_nearest
public static final int mkldnn_round_down
public static final int mkldnn_format_undef
public static final int mkldnn_any
public static final int mkldnn_blocked
public static final int mkldnn_x
public static final int mkldnn_nc
public static final int mkldnn_ncw
public static final int mkldnn_nwc
public static final int mkldnn_nchw
public static final int mkldnn_nhwc
public static final int mkldnn_chwn
public static final int mkldnn_ncdhw
public static final int mkldnn_ndhwc
public static final int mkldnn_oi
public static final int mkldnn_io
public static final int mkldnn_oiw
public static final int mkldnn_owi
public static final int mkldnn_wio
public static final int mkldnn_oihw
public static final int mkldnn_hwio
public static final int mkldnn_ohwi
public static final int mkldnn_ihwo
public static final int mkldnn_iohw
public static final int mkldnn_oidhw
public static final int mkldnn_dhwio
public static final int mkldnn_odhwi
public static final int mkldnn_goiw
public static final int mkldnn_goihw
public static final int mkldnn_hwigo
public static final int mkldnn_giohw
public static final int mkldnn_goidhw
public static final int mkldnn_ntc
public static final int mkldnn_tnc
public static final int mkldnn_ldsnc
public static final int mkldnn_ldigo
public static final int mkldnn_ldgoi
public static final int mkldnn_ldgo
public static final int mkldnn_nCw4c
public static final int mkldnn_nCw8c
public static final int mkldnn_nCw16c
public static final int mkldnn_nChw4c
public static final int mkldnn_nChw8c
public static final int mkldnn_nChw16c
public static final int mkldnn_nCdhw4c
public static final int mkldnn_nCdhw8c
public static final int mkldnn_nCdhw16c
public static final int mkldnn_Owi4o
public static final int mkldnn_OIw4i4o
public static final int mkldnn_Owi8o
public static final int mkldnn_OIw8i8o
public static final int mkldnn_OIw8o8i
public static final int mkldnn_OIw16i16o
public static final int mkldnn_OIw16o16i
public static final int mkldnn_Oiw4o
public static final int mkldnn_Oiw16o
public static final int mkldnn_Owi16o
public static final int mkldnn_OIw8i16o2i
public static final int mkldnn_OIw8o16i2o
public static final int mkldnn_IOw8o16i2o
public static final int mkldnn_IOw16o16i
public static final int mkldnn_OIw4i16o4i
public static final int mkldnn_OIw4i16o4i_s8s8
public static final int mkldnn_hwio_s8s8
public static final int mkldnn_oIhw8i
public static final int mkldnn_oIhw16i
public static final int mkldnn_OIhw4i4o
public static final int mkldnn_OIhw8i8o
public static final int mkldnn_OIhw16i16o
public static final int mkldnn_OIhw4i16o4i
public static final int mkldnn_OIhw4i16o4i_s8s8
public static final int mkldnn_OIhw8i16o2i
public static final int mkldnn_IOhw8i16o2i
public static final int mkldnn_OIhw8o16i2o
public static final int mkldnn_IOhw8o16i2o
public static final int mkldnn_OIhw8o8i
public static final int mkldnn_OIhw16o16i
public static final int mkldnn_IOhw16o16i
public static final int mkldnn_Oihw8o
public static final int mkldnn_Oihw4o
public static final int mkldnn_Oihw16o
public static final int mkldnn_Ohwi8o
public static final int mkldnn_Ohwi4o
public static final int mkldnn_Ohwi16o
public static final int mkldnn_OhIw16o4i
public static final int mkldnn_oIdhw8i
public static final int mkldnn_oIdhw16i
public static final int mkldnn_OIdhw4i4o
public static final int mkldnn_Odhwi4o
public static final int mkldnn_OIdhw8i8o
public static final int mkldnn_OIdhw8o8i
public static final int mkldnn_Odhwi8o
public static final int mkldnn_OIdhw16i16o
public static final int mkldnn_OIdhw16o16i
public static final int mkldnn_Oidhw4o
public static final int mkldnn_Oidhw16o
public static final int mkldnn_Odhwi16o
public static final int mkldnn_OIdhw8i16o2i
public static final int mkldnn_OIdhw8o16i2o
public static final int mkldnn_IOdhw8o16i2o
public static final int mkldnn_gOwi4o
public static final int mkldnn_gOIw4i4o
public static final int mkldnn_gOwi8o
public static final int mkldnn_gOIw8o8i
public static final int mkldnn_gOIw8i8o
public static final int mkldnn_gOIw16i16o
public static final int mkldnn_gOIw16o16i
public static final int mkldnn_gOiw4o
public static final int mkldnn_gOiw16o
public static final int mkldnn_gOwi16o
public static final int mkldnn_gOIw8i16o2i
public static final int mkldnn_gOIw8o16i2o
public static final int mkldnn_gIOw8o16i2o
public static final int mkldnn_gIOw16o16i
public static final int mkldnn_gOIw4i16o4i
public static final int mkldnn_gOIw4i16o4i_s8s8
public static final int mkldnn_Goiw16g
public static final int mkldnn_Goiw16g_s8s8
public static final int mkldnn_hwigo_s8s8
public static final int mkldnn_gOIhw4i4o
public static final int mkldnn_gOIhw8i8o
public static final int mkldnn_gOIhw16i16o
public static final int mkldnn_gOIhw4i16o4i
public static final int mkldnn_gOIhw4i16o4i_s8s8
public static final int mkldnn_gOIhw2i8o4i
public static final int mkldnn_gOIhw2i8o4i_s8s8
public static final int mkldnn_gOIhw8i16o2i
public static final int mkldnn_gIOhw8i16o2i
public static final int mkldnn_gOIhw8o16i2o
public static final int mkldnn_gIOhw8o16i2o
public static final int mkldnn_gOIhw4o4i
public static final int mkldnn_gOIhw4o4i_s8s8
public static final int mkldnn_gOIhw8o8i
public static final int mkldnn_gOIhw16o16i
public static final int mkldnn_gIOhw16o16i
public static final int mkldnn_gOihw8o
public static final int mkldnn_gOihw4o
public static final int mkldnn_gOihw16o
public static final int mkldnn_gOhwi8o
public static final int mkldnn_gOhwi4o
public static final int mkldnn_gOhwi16o
public static final int mkldnn_Goihw8g
public static final int mkldnn_Goihw16g
public static final int mkldnn_Goihw16g_s8s8
public static final int mkldnn_gOhIw16o4i
public static final int mkldnn_gOIdhw4i4o
public static final int mkldnn_gOdhwi4o
public static final int mkldnn_gOIdhw8i8o
public static final int mkldnn_gOIdhw8o8i
public static final int mkldnn_gOdhwi8o
public static final int mkldnn_gOIdhw8i16o2i
public static final int mkldnn_gOIdhw8o16i2o
public static final int mkldnn_gIOdhw8o16i2o
public static final int mkldnn_gOIdhw16i16o
public static final int mkldnn_gOIdhw16o16i
public static final int mkldnn_gOidhw4o
public static final int mkldnn_gOidhw16o
public static final int mkldnn_gOdhwi16o
public static final int mkldnn_wino_fmt
public static final int mkldnn_rnn_packed
public static final int mkldnn_format_last
public static final int mkldnn_padding_zero
public static final int mkldnn_prop_kind_undef
public static final int mkldnn_forward_training
public static final int mkldnn_forward_inference
public static final int mkldnn_forward_scoring
public static final int mkldnn_forward
public static final int mkldnn_backward
public static final int mkldnn_backward_data
public static final int mkldnn_backward_weights
public static final int mkldnn_backward_bias
public static final int mkldnn_undefined_primitive
public static final int mkldnn_memory
public static final int mkldnn_view
public static final int mkldnn_reorder
public static final int mkldnn_shuffle
public static final int mkldnn_concat
public static final int mkldnn_concat_inplace
public static final int mkldnn_sum
public static final int mkldnn_convolution
public static final int mkldnn_deconvolution
public static final int mkldnn_eltwise
public static final int mkldnn_softmax
public static final int mkldnn_pooling
public static final int mkldnn_lrn
public static final int mkldnn_batch_normalization
public static final int mkldnn_inner_product
public static final int mkldnn_rnn
public static final int mkldnn_alg_kind_undef
public static final int mkldnn_convolution_direct
public static final int mkldnn_convolution_winograd
public static final int mkldnn_convolution_auto
public static final int mkldnn_deconvolution_direct
public static final int mkldnn_deconvolution_winograd
public static final int mkldnn_eltwise_relu
public static final int mkldnn_eltwise_tanh
public static final int mkldnn_eltwise_elu
public static final int mkldnn_eltwise_square
public static final int mkldnn_eltwise_abs
public static final int mkldnn_eltwise_sqrt
public static final int mkldnn_eltwise_linear
public static final int mkldnn_eltwise_bounded_relu
public static final int mkldnn_eltwise_soft_relu
public static final int mkldnn_eltwise_logistic
public static final int mkldnn_eltwise_exp
public static final int mkldnn_eltwise_gelu
public static final int mkldnn_pooling_max
public static final int mkldnn_pooling_avg_include_padding
public static final int mkldnn_pooling_avg_exclude_padding
public static final int mkldnn_pooling_avg
public static final int mkldnn_lrn_across_channels
public static final int mkldnn_lrn_within_channel
public static final int mkldnn_vanilla_rnn
public static final int mkldnn_vanilla_lstm
public static final int mkldnn_vanilla_gru
public static final int mkldnn_gru_linear_before_reset
public static final int mkldnn_use_global_stats
public static final int mkldnn_use_scaleshift
public static final int mkldnn_fuse_bn_relu
public static final int TENSOR_MAX_DIMS
public static final int mkldnn_wino_undef
public static final int mkldnn_wino_wei_aaOIoi
public static final int mkldnn_wino_wei_aaOio
public static final int mkldnn_wino_wei_aaOBiOo
public static final int mkldnn_wino_wei_OBaaIBOIio
public static final int mkldnn_packed_format_undef
public static final int mkldnn_ldigo_p
public static final int mkldnn_ldgoi_p
public static final int MKLDNN_RNN_MAX_N_PARTS
public static final int mkldnn_rnn_cell_with_relu
public static final int mkldnn_rnn_cell_with_clipping
public static final int mkldnn_unidirectional_left2right
public static final int mkldnn_unidirectional_right2left
public static final int mkldnn_bidirectional_concat
public static final int mkldnn_bidirectional_sum
public static final int mkldnn_unidirectional
public static final int mkldnn_any_engine
public static final int mkldnn_cpu
public static final int mkldnn_query_undef
public static final int mkldnn_query_engine
public static final int mkldnn_query_primitive_kind
public static final int mkldnn_query_num_of_inputs_s32
public static final int mkldnn_query_num_of_outputs_s32
public static final int mkldnn_query_time_estimate_f64
public static final int mkldnn_query_memory_consumption_s64
public static final int mkldnn_query_impl_info_str
public static final int mkldnn_query_some_d
public static final int mkldnn_query_op_d
public static final int mkldnn_query_memory_d
public static final int mkldnn_query_convolution_d
public static final int mkldnn_query_deconvolution_d
public static final int mkldnn_query_shuffle_d
public static final int mkldnn_query_eltwise_d
public static final int mkldnn_query_softmax_d
public static final int mkldnn_query_pooling_d
public static final int mkldnn_query_lrn_d
public static final int mkldnn_query_batch_normalization_d
public static final int mkldnn_query_inner_product_d
public static final int mkldnn_query_rnn_d
public static final int mkldnn_query_some_pd
public static final int mkldnn_query_input_pd
public static final int mkldnn_query_output_pd
public static final int mkldnn_query_src_pd
public static final int mkldnn_query_diff_src_pd
public static final int mkldnn_query_weights_pd
public static final int mkldnn_query_diff_weights_pd
public static final int mkldnn_query_dst_pd
public static final int mkldnn_query_diff_dst_pd
public static final int mkldnn_query_workspace_pd
public static final int mkldnn_any_stream
public static final int mkldnn_eager
public static final int mkldnn_lazy
public static final int mkldnn_eager_nostore
public static final int round_nearest
public static final int round_down
public static final int zero
public static final int forward_training
public static final int forward_scoring
public static final int forward_inference
public static final int forward
public static final int backward
public static final int backward_data
public static final int backward_weights
public static final int backward_bias
public static final int algorithm_undef
public static final int convolution_auto
public static final int convolution_direct
public static final int convolution_winograd
public static final int deconvolution_direct
public static final int deconvolution_winograd
public static final int eltwise_relu
public static final int eltwise_tanh
public static final int eltwise_elu
public static final int eltwise_square
public static final int eltwise_abs
public static final int eltwise_sqrt
public static final int eltwise_linear
public static final int eltwise_bounded_relu
public static final int eltwise_soft_relu
public static final int eltwise_logistic
public static final int eltwise_exp
public static final int eltwise_gelu
public static final int lrn_across_channels
public static final int lrn_within_channel
public static final int pooling_max
public static final int pooling_avg
public static final int pooling_avg_include_padding
public static final int pooling_avg_exclude_padding
public static final int vanilla_rnn
public static final int vanilla_lstm
public static final int vanilla_gru
public static final int gru_linear_before_reset
public static final int use_global_stats
public static final int use_scale_shift
public static final int fuse_bn_relu
public static final int unidirectional_left2right
public static final int unidirectional_right2left
public static final int unidirectional
public static final int bidirectional_concat
public static final int bidirectional_sum
public static final int undef
public static final int eengine
public static final int primitive_kind
public static final int num_of_inputs_s32
public static final int num_of_outputs_s32
public static final int time_estimate_f64
public static final int memory_consumption_s64
public static final int impl_info_str
public static final int op_d
public static final int memory_d
public static final int convolution_d
public static final int deconvolution_d
public static final int shuffle_d
public static final int eltwise_d
public static final int softmax_d
public static final int pooling_d
public static final int lrn_d
public static final int batch_normalization_d
public static final int inner_product_d
public static final int rnn_d
public static final int input_pd
public static final int output_pd
public static final int src_pd
public static final int diff_src_pd
public static final int weights_pd
public static final int diff_weights_pd
public static final int dst_pd
public static final int diff_dst_pd
public static final int workspace_pd
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_create(@ByPtrPtr mkldnn_primitive_desc_iterator iterator, const_mkldnn_op_desc_t op_desc, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_create(@Cast(value="mkldnn_primitive_desc_iterator_t*") PointerPointer iterator, const_mkldnn_op_desc_t op_desc, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_create_v2(@ByPtrPtr mkldnn_primitive_desc_iterator iterator, const_mkldnn_op_desc_t op_desc, @Const mkldnn_primitive_attr attr, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_create_v2(@Cast(value="mkldnn_primitive_desc_iterator_t*") PointerPointer iterator, const_mkldnn_op_desc_t op_desc, @Const mkldnn_primitive_attr attr, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_next(mkldnn_primitive_desc_iterator iterator)
public static mkldnn_primitive_desc mkldnn_primitive_desc_iterator_fetch(@Const mkldnn_primitive_desc_iterator iterator)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_iterator_destroy(mkldnn_primitive_desc_iterator iterator)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc primitive_desc, const_mkldnn_op_desc_t op_desc, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer primitive_desc, const_mkldnn_op_desc_t op_desc, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_create_v2(@ByPtrPtr mkldnn_primitive_desc primitive_desc, const_mkldnn_op_desc_t op_desc, @Const mkldnn_primitive_attr attr, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_create_v2(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer primitive_desc, const_mkldnn_op_desc_t op_desc, @Const mkldnn_primitive_attr attr, mkldnn_engine engine, @Const mkldnn_primitive_desc hint_forward_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_clone(@ByPtrPtr mkldnn_primitive_desc primitive_desc, @Const mkldnn_primitive_desc existing_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_clone(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer primitive_desc, @Const mkldnn_primitive_desc existing_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_get_attr(@Const mkldnn_primitive_desc primitive_desc, @Const @ByPtrPtr mkldnn_primitive_attr attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_get_attr(@Const mkldnn_primitive_desc primitive_desc, @Cast(value="const_mkldnn_primitive_attr_t*") PointerPointer attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_destroy(mkldnn_primitive_desc primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_desc_query(@Const mkldnn_primitive_desc primitive_desc, @Cast(value="mkldnn_query_t") int what, int index, Pointer result)
for more options
@Const public static mkldnn_memory_desc_t mkldnn_primitive_desc_query_memory_d(@Const mkldnn_primitive_desc primitive_desc)
@Const public static mkldnn_primitive_desc mkldnn_primitive_desc_query_pd(@Const mkldnn_primitive_desc primitive_desc, @Cast(value="mkldnn_query_t") int what, int index)
public static int mkldnn_primitive_desc_query_s32(@Const mkldnn_primitive_desc primitive_desc, @Cast(value="mkldnn_query_t") int what, int index)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_create(@ByPtrPtr mkldnn_primitive primitive, @Const mkldnn_primitive_desc primitive_desc, @Const mkldnn_primitive_at_t inputs, @Const @ByPtrPtr mkldnn_primitive outputs)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_create(@Cast(value="mkldnn_primitive_t*") PointerPointer primitive, @Const mkldnn_primitive_desc primitive_desc, @Const mkldnn_primitive_at_t inputs, @Cast(value="const_mkldnn_primitive_t*") PointerPointer outputs)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_get_primitive_desc(@Const mkldnn_primitive primitive, @Const @ByPtrPtr mkldnn_primitive_desc primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_get_primitive_desc(@Const mkldnn_primitive primitive, @Cast(value="const_mkldnn_primitive_desc_t*") PointerPointer primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_get_input_at(@Const mkldnn_primitive primitive, @Cast(value="size_t") long index, mkldnn_primitive_at_t input)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_get_output(@Const mkldnn_primitive primitive, @Cast(value="size_t") long index, @Const @ByPtrPtr mkldnn_primitive output)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_get_output(@Const mkldnn_primitive primitive, @Cast(value="size_t") long index, @Cast(value="const_mkldnn_primitive_t*") PointerPointer output)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_destroy(mkldnn_primitive primitive)
@ByVal public static mkldnn_primitive_at_t mkldnn_primitive_at(@Const mkldnn_primitive primitive, @Cast(value="size_t") long output_index)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_create(@ByPtrPtr mkldnn_primitive_attr attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_create(@Cast(value="mkldnn_primitive_attr_t*") PointerPointer attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_clone(@ByPtrPtr mkldnn_primitive_attr attr, @Const mkldnn_primitive_attr existing_attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_clone(@Cast(value="mkldnn_primitive_attr_t*") PointerPointer attr, @Const mkldnn_primitive_attr existing_attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_destroy(mkldnn_primitive_attr attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_int_output_round_mode(@Const mkldnn_primitive_attr attr, @Cast(value="mkldnn_round_mode_t*") IntPointer round_mode)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_int_output_round_mode(@Const mkldnn_primitive_attr attr, @Cast(value="mkldnn_round_mode_t*") IntBuffer round_mode)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_int_output_round_mode(@Const mkldnn_primitive_attr attr, @Cast(value="mkldnn_round_mode_t*") int[] round_mode)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_int_output_round_mode(mkldnn_primitive_attr attr, @Cast(value="mkldnn_round_mode_t") int round_mode)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_output_scales(@Const mkldnn_primitive_attr attr, IntPointer count, IntPointer mask, @Cast(value="const float**") PointerPointer scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_output_scales(@Const mkldnn_primitive_attr attr, IntPointer count, IntPointer mask, @Const @ByPtrPtr FloatPointer scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_output_scales(@Const mkldnn_primitive_attr attr, IntBuffer count, IntBuffer mask, @Const @ByPtrPtr FloatBuffer scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_output_scales(@Const mkldnn_primitive_attr attr, int[] count, int[] mask, @Const @ByPtrPtr float[] scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr, int count, int mask, @Const FloatPointer scales)
int mb = 32, oc = 32, oh = 14, ow = 14; // convolution output params
float scales[oc] = { ... }; // unique output scales per output channel
int oc_dim = 1; // mb_dim = 0, channel_dim = 1, height_dim = 2, ...
mkldnn_convolution_desc_t cd; // create & configure convolution op_desc
mkldnn_primitive_attr_t attr;
mkldnn_primitive_attr_create(&attr); // create default attributes
mkldnn_primitive_attr_set_output_scales(attr, oc, 1 << oc_dim, scales);
mkldnn_primitive_desc_t cpd;
mkldnn_primitive_desc_create_v2(&cpd, &cd, attr, NULL);
\note
There is no way to check that \p count corresponds to \p mask until an
actual primitive descriptor is created, so it is the user's
responsibility to set proper values. The following formula must hold:
\[count = \prod\limits_{d \in mask} output.dims[d]\]
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr, int count, int mask, @Const FloatBuffer scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr, int count, int mask, @Const float[] scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_post_ops(@Const mkldnn_primitive_attr attr, @Const @ByPtrPtr mkldnn_post_ops post_ops)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_get_post_ops(@Const mkldnn_primitive_attr attr, @Cast(value="const_mkldnn_post_ops_t*") PointerPointer post_ops)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_post_ops(mkldnn_primitive_attr attr, @Const mkldnn_post_ops post_ops)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_create(@ByPtrPtr mkldnn_post_ops post_ops)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_create(@Cast(value="mkldnn_post_ops_t*") PointerPointer post_ops)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_destroy(mkldnn_post_ops post_ops)
public static int mkldnn_post_ops_len(@Const mkldnn_post_ops post_ops)
@Cast(value="mkldnn_primitive_kind_t") public static int mkldnn_post_ops_get_kind(@Const mkldnn_post_ops post_ops, int index)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_append_sum(mkldnn_post_ops post_ops, float scale)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_sum(@Const mkldnn_post_ops post_ops, int index, FloatPointer scale)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_sum(@Const mkldnn_post_ops post_ops, int index, FloatBuffer scale)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_sum(@Const mkldnn_post_ops post_ops, int index, float[] scale)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_append_eltwise(mkldnn_post_ops post_ops, float scale, @Cast(value="mkldnn_alg_kind_t") int alg, float alpha, float beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_eltwise(@Const mkldnn_post_ops post_ops, int index, FloatPointer scale, @Cast(value="mkldnn_alg_kind_t*") IntPointer alg, FloatPointer alpha, FloatPointer beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_eltwise(@Const mkldnn_post_ops post_ops, int index, FloatBuffer scale, @Cast(value="mkldnn_alg_kind_t*") IntBuffer alg, FloatBuffer alpha, FloatBuffer beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_post_ops_get_params_eltwise(@Const mkldnn_post_ops post_ops, int index, float[] scale, @Cast(value="mkldnn_alg_kind_t*") int[] alg, float[] alpha, float[] beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc, int ndims, @Const IntPointer dims, @Cast(value="mkldnn_data_type_t") int data_type, @Cast(value="mkldnn_memory_format_t") int format)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc, int ndims, @Const IntBuffer dims, @Cast(value="mkldnn_data_type_t") int data_type, @Cast(value="mkldnn_memory_format_t") int format)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_desc_init(mkldnn_memory_desc_t memory_desc, int ndims, @Const int[] dims, @Cast(value="mkldnn_data_type_t") int data_type, @Cast(value="mkldnn_memory_format_t") int format)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc memory_primitive_desc, @Const mkldnn_memory_desc_t memory_desc, mkldnn_engine engine)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer memory_primitive_desc, @Const mkldnn_memory_desc_t memory_desc, mkldnn_engine engine)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const IntPointer dims, @Const IntPointer offsets)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const IntBuffer dims, @Const IntBuffer offsets)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const int[] dims, @Const int[] offsets)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const IntPointer dims, @Const IntPointer offsets)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const IntBuffer dims, @Const IntBuffer offsets)
@Cast(value="mkldnn_status_t") public static int mkldnn_view_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer view_primitive_desc, @Const mkldnn_primitive_desc memory_primitive_desc, @Const int[] dims, @Const int[] offsets)
public static int mkldnn_memory_primitive_desc_equal(@Const mkldnn_primitive_desc lhs, @Const mkldnn_primitive_desc rhs)
@Cast(value="size_t") public static long mkldnn_memory_primitive_desc_get_size(@Const mkldnn_primitive_desc memory_primitive_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_get_data_handle(@Const mkldnn_primitive memory, @Cast(value="void**") PointerPointer handle)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_get_data_handle(@Const mkldnn_primitive memory, @Cast(value="void**") @ByPtrPtr Pointer handle)
@Cast(value="mkldnn_status_t") public static int mkldnn_memory_set_data_handle(mkldnn_primitive memory, Pointer handle)
@Cast(value="mkldnn_status_t") public static int mkldnn_reorder_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc reorder_primitive_desc, @Const mkldnn_primitive_desc input, @Const mkldnn_primitive_desc output)
@Cast(value="mkldnn_status_t") public static int mkldnn_reorder_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer reorder_primitive_desc, @Const mkldnn_primitive_desc input, @Const mkldnn_primitive_desc output)
@Cast(value="mkldnn_status_t") public static int mkldnn_reorder_primitive_desc_create_v2(@ByPtrPtr mkldnn_primitive_desc reorder_primitive_desc, @Const mkldnn_primitive_desc input, @Const mkldnn_primitive_desc output, @Const mkldnn_primitive_attr attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_reorder_primitive_desc_create_v2(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer reorder_primitive_desc, @Const mkldnn_primitive_desc input, @Const mkldnn_primitive_desc output, @Const mkldnn_primitive_attr attr)
@Cast(value="mkldnn_status_t") public static int mkldnn_concat_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc concat_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, int concat_dimension, @Const @ByPtrPtr mkldnn_primitive_desc input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_concat_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer concat_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, int concat_dimension, @Cast(value="const_mkldnn_primitive_desc_t*") PointerPointer input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const FloatPointer scales, @Const @ByPtrPtr mkldnn_primitive_desc input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const FloatBuffer scales, @Cast(value="const_mkldnn_primitive_desc_t*") PointerPointer input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const float[] scales, @Const @ByPtrPtr mkldnn_primitive_desc input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const FloatPointer scales, @Cast(value="const_mkldnn_primitive_desc_t*") PointerPointer input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@ByPtrPtr mkldnn_primitive_desc sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const FloatBuffer scales, @Const @ByPtrPtr mkldnn_primitive_desc input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_sum_primitive_desc_create(@Cast(value="mkldnn_primitive_desc_t*") PointerPointer sum_primitive_desc, @Const mkldnn_memory_desc_t output_desc, int n, @Const float[] scales, @Cast(value="const_mkldnn_primitive_desc_t*") PointerPointer input_pds)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_forward_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_data_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer dilates, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer dilates, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_dilated_deconvolution_backward_weights_desc_init(@Cast(value="mkldnn_deconvolution_desc_t*") mkldnn_convolution_desc_t conv_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] dilates, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_shuffle_forward_desc_init(mkldnn_shuffle_desc_t shuffle_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_memory_desc_t data_desc, int axis, int group_size)
@Cast(value="mkldnn_status_t") public static int mkldnn_shuffle_backward_desc_init(mkldnn_shuffle_desc_t shuffle_desc, @Const mkldnn_memory_desc_t diff_data_desc, int axis, int group_size)
@Cast(value="mkldnn_status_t") public static int mkldnn_eltwise_forward_desc_init(mkldnn_eltwise_desc_t eltwise_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t data_desc, float alpha, float beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_eltwise_backward_desc_init(mkldnn_eltwise_desc_t eltwise_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_data_desc, @Const mkldnn_memory_desc_t data_desc, float alpha, float beta)
@Cast(value="mkldnn_status_t") public static int mkldnn_softmax_forward_desc_init(mkldnn_softmax_desc_t softmax_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_memory_desc_t data_desc, int softmax_axis)
@Cast(value="mkldnn_status_t") public static int mkldnn_softmax_backward_desc_init(mkldnn_softmax_desc_t softmax_desc, @Const mkldnn_memory_desc_t diff_desc, @Const mkldnn_memory_desc_t data_desc, int softmax_axis)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntPointer strides, @Const IntPointer kernel, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t dst_desc, @Const IntBuffer strides, @Const IntBuffer kernel, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t dst_desc, @Const int[] strides, @Const int[] kernel, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntPointer strides, @Const IntPointer kernel, @Const IntPointer padding_l, @Const IntPointer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const IntBuffer strides, @Const IntBuffer kernel, @Const IntBuffer padding_l, @Const IntBuffer padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t pool_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t diff_dst_desc, @Const int[] strides, @Const int[] kernel, @Const int[] padding_l, @Const int[] padding_r, @Cast(value="mkldnn_padding_kind_t") int padding_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_lrn_forward_desc_init(mkldnn_lrn_desc_t lrn_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t data_desc, int local_size, float alpha, float beta, float k)
@Cast(value="mkldnn_status_t") public static int mkldnn_lrn_backward_desc_init(mkldnn_lrn_desc_t lrn_desc, @Cast(value="mkldnn_alg_kind_t") int alg_kind, @Const mkldnn_memory_desc_t diff_data_desc, @Const mkldnn_memory_desc_t data_desc, int local_size, float alpha, float beta, float k)
@Cast(value="mkldnn_status_t") public static int mkldnn_batch_normalization_forward_desc_init(mkldnn_batch_normalization_desc_t bnrm_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_memory_desc_t data_desc, float epsilon, @Cast(value="unsigned") int flags)
mkldnn_batch_normalization_desc_t
@Cast(value="mkldnn_status_t") public static int mkldnn_batch_normalization_backward_desc_init(mkldnn_batch_normalization_desc_t bnrm_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_memory_desc_t diff_data_desc, @Const mkldnn_memory_desc_t data_desc, float epsilon, @Cast(value="unsigned") int flags)
mkldnn_batch_normalization_desc_t
@Cast(value="mkldnn_status_t") public static int mkldnn_inner_product_forward_desc_init(mkldnn_inner_product_desc_t ip_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_inner_product_backward_data_desc_init(mkldnn_inner_product_desc_t ip_desc, @Const mkldnn_memory_desc_t diff_src_desc, @Const mkldnn_memory_desc_t weights_desc, @Const mkldnn_memory_desc_t diff_dst_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_inner_product_backward_weights_desc_init(mkldnn_inner_product_desc_t ip_desc, @Const mkldnn_memory_desc_t src_desc, @Const mkldnn_memory_desc_t diff_weights_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_rnn_cell_desc_init(mkldnn_rnn_cell_desc_t rnn_cell_desc, @Cast(value="mkldnn_alg_kind_t") int kind, @Cast(value="mkldnn_alg_kind_t") int f, @Cast(value="unsigned int") int flags, float alpha, float clipping)
public static int mkldnn_rnn_cell_get_gates_count(@Const mkldnn_rnn_cell_desc_t rnn_cell_desc)
public static int mkldnn_rnn_cell_get_states_count(@Const mkldnn_rnn_cell_desc_t rnn_cell_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_rnn_data_qparams(mkldnn_primitive_attr attr, float scale, float shift)
// rnn parameters
int l = 2, t = 2, mb = 32, sic = 32, slc = 32, dic = 32, dlc = 32;
// activations quantization parameters
float scale = ..., shift = ..;
mkldnn_primitive_attr_t rnn_attr;
// create default attributes
mkldnn_primitive_attr_create(&rnn_attr);
// set scale and shift for int8 quantization of activation
mkldnn_primitive_attr_set_rnn_data_qparams(rnn_attr, scale, shift);
// create & configure rnn op_desc
mkldnn_rnn_desc_t rnn_d;
mkldnn_primitive_desc_t rnn_pd;
mkldnn_primitive_desc_create_v2(&rnn_pd, &rnn_d, attr, NULL);
\note
Quantization scale and shift are common for src_layer, src_iter,
dst_iter and dst_layer.@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr, int count, int mask, @Const FloatPointer weights_scales)
// rnn parameters
int l = 2, t = 2, mb = 32, sic = 32, slc = 32, dic = 32, dlc = 32;
// unique output scales per output channel
float weights_scales[dic * n_gates] = { ... };
// mask that specifies last two dimensions of ldigo format
int mask = 0x3;
mkldnn_primitive_attr_t attr;
// create default attributes
mkldnn_primitive_attr_create(&attr);
// set output channel-wise weights scales
mkldnn_primitive_attr_set_rnn_weights_qparams(attr, dic * n_gates, mask,
weights_scales);
// create & configure rnn op_desc
mkldnn_rnn_desc_t rnn_d;
mkldnn_primitive_desc_t rnn_pd;
mkldnn_primitive_desc_create_v2(&rnn_pd, &rnn_d, attr, NULL);
\note
The dimension order is always native and does not depend on the actual
layout used. For example, 5 dimensional weights always have
(l, d, i, g, o) logical dimension ordering.
\note
Quantization sales are common for weights_layer and weights_iteration
\note
There is no way to check that \p count corresponds to \p mask until an
actual primitive descriptor is created, so it is user's responsibility
to set proper values. The following formula must be held:
\[count = \prod\limits_{d \in mask} output.dims[d]\]
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr, int count, int mask, @Const FloatBuffer weights_scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr, int count, int mask, @Const float[] weights_scales)
@Cast(value="mkldnn_status_t") public static int mkldnn_rnn_forward_desc_init(mkldnn_rnn_desc_t rnn_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_rnn_cell_desc_t rnn_cell_desc, @Cast(value="const mkldnn_rnn_direction_t") int direction, @Const mkldnn_memory_desc_t src_layer_desc, @Const mkldnn_memory_desc_t src_iter_desc, @Const mkldnn_memory_desc_t weights_layer_desc, @Const mkldnn_memory_desc_t weights_iter_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_layer_desc, @Const mkldnn_memory_desc_t dst_iter_desc)
@Cast(value="mkldnn_status_t") public static int mkldnn_rnn_backward_desc_init(mkldnn_rnn_desc_t rnn_desc, @Cast(value="mkldnn_prop_kind_t") int prop_kind, @Const mkldnn_rnn_cell_desc_t rnn_cell_desc, @Cast(value="const mkldnn_rnn_direction_t") int direction, @Const mkldnn_memory_desc_t src_layer_desc, @Const mkldnn_memory_desc_t src_iter_desc, @Const mkldnn_memory_desc_t weights_layer_desc, @Const mkldnn_memory_desc_t weights_iter_desc, @Const mkldnn_memory_desc_t bias_desc, @Const mkldnn_memory_desc_t dst_layer_desc, @Const mkldnn_memory_desc_t dst_iter_desc, @Const mkldnn_memory_desc_t diff_src_layer_desc, @Const mkldnn_memory_desc_t diff_src_iter_desc, @Const mkldnn_memory_desc_t diff_weights_layer_desc, @Const mkldnn_memory_desc_t diff_weights_iter_desc, @Const mkldnn_memory_desc_t diff_bias_desc, @Const mkldnn_memory_desc_t diff_dst_layer, @Const mkldnn_memory_desc_t diff_dst_iter_desc)
@Cast(value="size_t") public static long mkldnn_engine_get_count(@Cast(value="mkldnn_engine_kind_t") int kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_create(@ByPtrPtr mkldnn_engine engine, @Cast(value="mkldnn_engine_kind_t") int kind, @Cast(value="size_t") long index)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_create(@Cast(value="mkldnn_engine_t*") PointerPointer engine, @Cast(value="mkldnn_engine_kind_t") int kind, @Cast(value="size_t") long index)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_get_kind(mkldnn_engine engine, @Cast(value="mkldnn_engine_kind_t*") IntPointer kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_get_kind(mkldnn_engine engine, @Cast(value="mkldnn_engine_kind_t*") IntBuffer kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_get_kind(mkldnn_engine engine, @Cast(value="mkldnn_engine_kind_t*") int[] kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_engine_destroy(mkldnn_engine engine)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_create(@ByPtrPtr mkldnn_stream stream, @Cast(value="mkldnn_stream_kind_t") int stream_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_create(@Cast(value="mkldnn_stream_t*") PointerPointer stream, @Cast(value="mkldnn_stream_kind_t") int stream_kind)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_submit(mkldnn_stream stream, @Cast(value="size_t") long n, @ByPtrPtr mkldnn_primitive primitives, @ByPtrPtr mkldnn_primitive error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_submit(mkldnn_stream stream, @Cast(value="size_t") long n, @Cast(value="mkldnn_primitive_t*") PointerPointer primitives, @Cast(value="mkldnn_primitive_t*") PointerPointer error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_wait(mkldnn_stream stream, int block, @ByPtrPtr mkldnn_primitive error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_wait(mkldnn_stream stream, int block, @Cast(value="mkldnn_primitive_t*") PointerPointer error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_rerun(mkldnn_stream stream, @ByPtrPtr mkldnn_primitive error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_rerun(mkldnn_stream stream, @Cast(value="mkldnn_primitive_t*") PointerPointer error_primitive)
@Cast(value="mkldnn_status_t") public static int mkldnn_stream_destroy(mkldnn_stream stream)
@Cast(value="mkldnn_status_t") public static int mkldnn_set_verbose(int level)
@Cast(value="mkldnn_status_t") public static int mkldnn_set_jit_dump(int dump)
@Const public static mkldnn_version_t mkldnn_version()
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const FloatPointer A, @Const IntPointer lda, @Const FloatPointer B, @Const IntPointer ldb, @Const FloatPointer beta, FloatPointer C, @Const IntPointer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(String transa, String transb, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const FloatBuffer A, @Const IntBuffer lda, @Const FloatBuffer B, @Const IntBuffer ldb, @Const FloatBuffer beta, FloatBuffer C, @Const IntBuffer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const float[] A, @Const int[] lda, @Const float[] B, @Const int[] ldb, @Const float[] beta, float[] C, @Const int[] ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(String transa, String transb, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const FloatPointer A, @Const IntPointer lda, @Const FloatPointer B, @Const IntPointer ldb, @Const FloatPointer beta, FloatPointer C, @Const IntPointer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const FloatBuffer A, @Const IntBuffer lda, @Const FloatBuffer B, @Const IntBuffer ldb, @Const FloatBuffer beta, FloatBuffer C, @Const IntBuffer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_sgemm(String transa, String transb, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const float[] A, @Const int[] lda, @Const float[] B, @Const int[] ldb, @Const float[] beta, float[] C, @Const int[] ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const BytePointer A, @Const IntPointer lda, @Const BytePointer ao, @Cast(value="const uint8_t*") BytePointer B, @Const IntPointer ldb, @Const BytePointer bo, @Const FloatPointer beta, IntPointer c, @Const IntPointer ldc, @Const IntPointer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(String transa, String transb, String offsetc, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const ByteBuffer A, @Const IntBuffer lda, @Const ByteBuffer ao, @Cast(value="const uint8_t*") ByteBuffer B, @Const IntBuffer ldb, @Const ByteBuffer bo, @Const FloatBuffer beta, IntBuffer c, @Const IntBuffer ldc, @Const IntBuffer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const byte[] A, @Const int[] lda, @Const byte[] ao, @Cast(value="const uint8_t*") byte[] B, @Const int[] ldb, @Const byte[] bo, @Const float[] beta, int[] c, @Const int[] ldc, @Const int[] co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(String transa, String transb, String offsetc, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const BytePointer A, @Const IntPointer lda, @Const BytePointer ao, @Cast(value="const uint8_t*") BytePointer B, @Const IntPointer ldb, @Const BytePointer bo, @Const FloatPointer beta, IntPointer c, @Const IntPointer ldc, @Const IntPointer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const ByteBuffer A, @Const IntBuffer lda, @Const ByteBuffer ao, @Cast(value="const uint8_t*") ByteBuffer B, @Const IntBuffer ldb, @Const ByteBuffer bo, @Const FloatBuffer beta, IntBuffer c, @Const IntBuffer ldc, @Const IntBuffer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8u8s32(String transa, String transb, String offsetc, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const byte[] A, @Const int[] lda, @Const byte[] ao, @Cast(value="const uint8_t*") byte[] B, @Const int[] ldb, @Const byte[] bo, @Const float[] beta, int[] c, @Const int[] ldc, @Const int[] co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const BytePointer A, @Const IntPointer lda, @Const BytePointer ao, @Const BytePointer B, @Const IntPointer ldb, @Const BytePointer bo, @Const FloatPointer beta, IntPointer c, @Const IntPointer ldc, @Const IntPointer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(String transa, String transb, String offsetc, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const ByteBuffer A, @Const IntBuffer lda, @Const ByteBuffer ao, @Const ByteBuffer B, @Const IntBuffer ldb, @Const ByteBuffer bo, @Const FloatBuffer beta, IntBuffer c, @Const IntBuffer ldc, @Const IntBuffer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const byte[] A, @Const int[] lda, @Const byte[] ao, @Const byte[] B, @Const int[] ldb, @Const byte[] bo, @Const float[] beta, int[] c, @Const int[] ldc, @Const int[] co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(String transa, String transb, String offsetc, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Const BytePointer A, @Const IntPointer lda, @Const BytePointer ao, @Const BytePointer B, @Const IntPointer ldb, @Const BytePointer bo, @Const FloatPointer beta, IntPointer c, @Const IntPointer ldc, @Const IntPointer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Cast(value="const char*") BytePointer offsetc, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Const ByteBuffer A, @Const IntBuffer lda, @Const ByteBuffer ao, @Const ByteBuffer B, @Const IntBuffer ldb, @Const ByteBuffer bo, @Const FloatBuffer beta, IntBuffer c, @Const IntBuffer ldc, @Const IntBuffer co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_s8s8s32(String transa, String transb, String offsetc, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Const byte[] A, @Const int[] lda, @Const byte[] ao, @Const byte[] B, @Const int[] ldb, @Const byte[] bo, @Const float[] beta, int[] c, @Const int[] ldc, @Const int[] co)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Cast(value="const mkldnn_bfloat16_t*") ShortPointer A, @Const IntPointer lda, @Cast(value="const mkldnn_bfloat16_t*") ShortPointer B, @Const IntPointer ldb, @Const FloatPointer beta, FloatPointer c, @Const IntPointer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(String transa, String transb, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Cast(value="const mkldnn_bfloat16_t*") ShortBuffer A, @Const IntBuffer lda, @Cast(value="const mkldnn_bfloat16_t*") ShortBuffer B, @Const IntBuffer ldb, @Const FloatBuffer beta, FloatBuffer c, @Const IntBuffer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Cast(value="const mkldnn_bfloat16_t*") short[] A, @Const int[] lda, @Cast(value="const mkldnn_bfloat16_t*") short[] B, @Const int[] ldb, @Const float[] beta, float[] c, @Const int[] ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(String transa, String transb, @Const IntPointer M, @Const IntPointer N, @Const IntPointer K, @Const FloatPointer alpha, @Cast(value="const mkldnn_bfloat16_t*") ShortPointer A, @Const IntPointer lda, @Cast(value="const mkldnn_bfloat16_t*") ShortPointer B, @Const IntPointer ldb, @Const FloatPointer beta, FloatPointer c, @Const IntPointer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(@Cast(value="const char*") BytePointer transa, @Cast(value="const char*") BytePointer transb, @Const IntBuffer M, @Const IntBuffer N, @Const IntBuffer K, @Const FloatBuffer alpha, @Cast(value="const mkldnn_bfloat16_t*") ShortBuffer A, @Const IntBuffer lda, @Cast(value="const mkldnn_bfloat16_t*") ShortBuffer B, @Const IntBuffer ldb, @Const FloatBuffer beta, FloatBuffer c, @Const IntBuffer ldc)
@Cast(value="mkldnn_status_t") public static int mkldnn_gemm_bf16bf16f32(String transa, String transb, @Const int[] M, @Const int[] N, @Const int[] K, @Const float[] alpha, @Cast(value="const mkldnn_bfloat16_t*") short[] A, @Const int[] lda, @Cast(value="const mkldnn_bfloat16_t*") short[] B, @Const int[] ldb, @Const float[] beta, float[] c, @Const int[] ldc)
@Namespace(value="mkldnn") @Cast(value="mkldnn_primitive_kind_t") public static int convert_to_c(@Cast(value="mkldnn::primitive::kind") int akind)
\addtogroup cpp_api_utils Utils \{
A class that provides the destructor for an Intel(R) MKL-DNN C handle
@Namespace(value="mkldnn") @ByVal public static memory.desc zero_md()
@Namespace(value="mkldnn") public static void check_num_parameters(@Const mkldnn_primitive_desc aprimitive_desc, int n_inputs, int n_outputs, @StdString BytePointer prim_name)
@Namespace(value="mkldnn") public static void check_num_parameters(@Const mkldnn_primitive_desc aprimitive_desc, int n_inputs, int n_outputs, @StdString String prim_name)
@Namespace(value="mkldnn") @Cast(value="bool") public static boolean is_null_memory(@Const mkldnn_primitive aprimitive)
@Namespace(value="mkldnn") @Cast(value="bool") @Name(value="operator ==") public static boolean equals(@Cast(value="mkldnn_data_type_t") int a, @Cast(value="mkldnn::memory::data_type") int b)
Copyright © 2022. All rights reserved.