|
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::convolution_direct = mkldnn_convolution_direct,
mkldnn::convolution_winograd = mkldnn_convolution_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::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
} |
|
enum | mkldnn::batch_normalization_flag { mkldnn::use_global_stats = mkldnn_use_global_stats,
mkldnn::use_scale_shift = mkldnn_use_scaleshift,
mkldnn::omit_stats = mkldnn_omit_stats,
mkldnn::fuse_bn_relu = mkldnn_fuse_bn_relu
} |
|
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::memory_d = mkldnn_query_memory_d,
mkldnn::convolution_d = mkldnn_query_convolution_d,
mkldnn::eltwise_d = mkldnn_query_eltwise_d,
mkldnn::relu_d = mkldnn_query_relu_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::convolution_relu_d = mkldnn_query_convolution_relu_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
} |
|