Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  0.21.0
Performance library for Deep Learning
Enumerations | Functions
Common data types and enumerations

A proxy to Types in C API. More...

Enumerations

enum  mkldnn::round_mode { mkldnn::round_nearest = mkldnn_round_nearest, mkldnn::round_down = mkldnn_round_down }
 
enum  mkldnn::padding_kind { mkldnn::zero = mkldnn_padding_zero }
 
enum  mkldnn::prop_kind {
  mkldnn::forward_training = mkldnn_forward_training, mkldnn::forward_scoring = mkldnn_forward_scoring, mkldnn::forward_inference = mkldnn_forward_inference, mkldnn::forward = mkldnn_forward,
  mkldnn::backward = mkldnn_backward, mkldnn::backward_data = mkldnn_backward_data, mkldnn::backward_weights = mkldnn_backward_weights, mkldnn::backward_bias = mkldnn_backward_bias
}
 
enum  mkldnn::algorithm {
  mkldnn::algorithm_undef = mkldnn_alg_kind_undef, mkldnn::convolution_auto = mkldnn_convolution_auto, mkldnn::convolution_direct = mkldnn_convolution_direct, mkldnn::convolution_winograd = mkldnn_convolution_winograd,
  mkldnn::deconvolution_direct = mkldnn_deconvolution_direct, mkldnn::deconvolution_winograd = mkldnn_deconvolution_winograd, mkldnn::eltwise_relu = mkldnn_eltwise_relu, mkldnn::eltwise_tanh = mkldnn_eltwise_tanh,
  mkldnn::eltwise_elu = mkldnn_eltwise_elu, mkldnn::eltwise_square = mkldnn_eltwise_square, mkldnn::eltwise_abs = mkldnn_eltwise_abs, mkldnn::eltwise_sqrt = mkldnn_eltwise_sqrt,
  mkldnn::eltwise_linear = mkldnn_eltwise_linear, mkldnn::eltwise_bounded_relu = mkldnn_eltwise_bounded_relu, mkldnn::eltwise_soft_relu = mkldnn_eltwise_soft_relu, mkldnn::eltwise_logistic = mkldnn_eltwise_logistic,
  mkldnn::eltwise_exp = mkldnn_eltwise_exp, mkldnn::eltwise_gelu = mkldnn_eltwise_gelu, mkldnn::lrn_across_channels = mkldnn_lrn_across_channels, mkldnn::lrn_within_channel = mkldnn_lrn_within_channel,
  mkldnn::pooling_max = mkldnn_pooling_max, mkldnn::pooling_avg = mkldnn_pooling_avg, mkldnn::pooling_avg_include_padding = mkldnn_pooling_avg_include_padding, mkldnn::pooling_avg_exclude_padding = mkldnn_pooling_avg_exclude_padding,
  mkldnn::vanilla_rnn = mkldnn_vanilla_rnn, mkldnn::vanilla_lstm = mkldnn_vanilla_lstm, mkldnn::vanilla_gru = mkldnn_vanilla_gru, mkldnn::gru_linear_before_reset = mkldnn_gru_linear_before_reset
}
 
enum  mkldnn::batch_normalization_flag { mkldnn::use_global_stats = mkldnn_use_global_stats, mkldnn::use_scale_shift = mkldnn_use_scaleshift, mkldnn::fuse_bn_relu = mkldnn_fuse_bn_relu }
 
enum  mkldnn::rnn_direction {
  mkldnn::unidirectional_left2right = mkldnn_unidirectional_left2right, mkldnn::unidirectional_right2left = mkldnn_unidirectional_right2left, mkldnn::unidirectional = mkldnn_unidirectional, mkldnn::bidirectional_concat = mkldnn_bidirectional_concat,
  mkldnn::bidirectional_sum = mkldnn_bidirectional_sum
}
 
enum  mkldnn::query {
  mkldnn::undef = mkldnn_query_undef, mkldnn::eengine = mkldnn_query_engine, mkldnn::primitive_kind = mkldnn_query_primitive_kind, mkldnn::num_of_inputs_s32 = mkldnn_query_num_of_inputs_s32,
  mkldnn::num_of_outputs_s32 = mkldnn_query_num_of_outputs_s32, mkldnn::time_estimate_f64 = mkldnn_query_time_estimate_f64, mkldnn::memory_consumption_s64 = mkldnn_query_memory_consumption_s64, mkldnn::impl_info_str = mkldnn_query_impl_info_str,
  mkldnn::op_d = mkldnn_query_op_d, mkldnn::memory_d = mkldnn_query_memory_d, mkldnn::convolution_d = mkldnn_query_convolution_d, mkldnn::deconvolution_d = mkldnn_query_deconvolution_d,
  mkldnn::shuffle_d = mkldnn_query_shuffle_d, mkldnn::eltwise_d = mkldnn_query_eltwise_d, mkldnn::softmax_d = mkldnn_query_softmax_d, mkldnn::pooling_d = mkldnn_query_pooling_d,
  mkldnn::lrn_d = mkldnn_query_lrn_d, mkldnn::batch_normalization_d = mkldnn_query_batch_normalization_d, mkldnn::inner_product_d = mkldnn_query_inner_product_d, mkldnn::rnn_d = mkldnn_query_rnn_d,
  mkldnn::input_pd = mkldnn_query_input_pd, mkldnn::output_pd = mkldnn_query_output_pd, mkldnn::src_pd = mkldnn_query_src_pd, mkldnn::diff_src_pd = mkldnn_query_diff_src_pd,
  mkldnn::weights_pd = mkldnn_query_weights_pd, mkldnn::diff_weights_pd = mkldnn_query_diff_weights_pd, mkldnn::dst_pd = mkldnn_query_dst_pd, mkldnn::diff_dst_pd = mkldnn_query_diff_dst_pd,
  mkldnn::workspace_pd = mkldnn_query_workspace_pd
}
 

Functions

mkldnn_round_mode_t mkldnn::convert_to_c (round_mode mode)
 
mkldnn_padding_kind_t mkldnn::convert_to_c (padding_kind kind)
 
mkldnn_prop_kind_t mkldnn::convert_to_c (prop_kind kind)
 
mkldnn_alg_kind_t mkldnn::convert_to_c (algorithm aalgorithm)
 
mkldnn_batch_normalization_flag_t mkldnn::convert_to_c (batch_normalization_flag aflag)
 
mkldnn_rnn_direction_t mkldnn::convert_to_c (rnn_direction adir)
 
mkldnn_query_t mkldnn::convert_to_c (query aquery)
 

Detailed Description

A proxy to Types in C API.

Enumeration Type Documentation

◆ round_mode

Enumerator
round_nearest 
round_down 

◆ padding_kind

Enumerator
zero 

◆ prop_kind

Enumerator
forward_training 
forward_scoring 
forward_inference 
forward 
backward 
backward_data 
backward_weights 
backward_bias 

◆ algorithm

Enumerator
algorithm_undef 
convolution_auto 
convolution_direct 
convolution_winograd 
deconvolution_direct 
deconvolution_winograd 
eltwise_relu 
eltwise_tanh 
eltwise_elu 
eltwise_square 
eltwise_abs 
eltwise_sqrt 
eltwise_linear 
eltwise_bounded_relu 
eltwise_soft_relu 
eltwise_logistic 
eltwise_exp 
eltwise_gelu 
lrn_across_channels 
lrn_within_channel 
pooling_max 
pooling_avg 
pooling_avg_include_padding 
pooling_avg_exclude_padding 
vanilla_rnn 
vanilla_lstm 
vanilla_gru 
gru_linear_before_reset 

◆ batch_normalization_flag

Enumerator
use_global_stats 
use_scale_shift 
fuse_bn_relu 

◆ rnn_direction

Enumerator
unidirectional_left2right 
unidirectional_right2left 
unidirectional 
bidirectional_concat 
bidirectional_sum 

◆ query

Enumerator
undef 
eengine 
primitive_kind 
num_of_inputs_s32 
num_of_outputs_s32 
time_estimate_f64 
memory_consumption_s64 
impl_info_str 
op_d 
memory_d 
convolution_d 
deconvolution_d 
shuffle_d 
eltwise_d 
softmax_d 
pooling_d 
lrn_d 
batch_normalization_d 
inner_product_d 
rnn_d 
input_pd 
output_pd 
src_pd 
diff_src_pd 
weights_pd 
diff_weights_pd 
dst_pd 
diff_dst_pd 
workspace_pd 

Function Documentation

◆ convert_to_c() [1/7]

mkldnn_round_mode_t mkldnn::convert_to_c ( round_mode  mode)
inline

◆ convert_to_c() [2/7]

mkldnn_padding_kind_t mkldnn::convert_to_c ( padding_kind  kind)
inline

◆ convert_to_c() [3/7]

mkldnn_prop_kind_t mkldnn::convert_to_c ( prop_kind  kind)
inline

◆ convert_to_c() [4/7]

mkldnn_alg_kind_t mkldnn::convert_to_c ( algorithm  aalgorithm)
inline

◆ convert_to_c() [5/7]

mkldnn_batch_normalization_flag_t mkldnn::convert_to_c ( batch_normalization_flag  aflag)
inline

◆ convert_to_c() [6/7]

mkldnn_rnn_direction_t mkldnn::convert_to_c ( rnn_direction  adir)
inline

◆ convert_to_c() [7/7]

mkldnn_query_t mkldnn::convert_to_c ( query  aquery)
inline