mkldnn.hpp File Reference

Go to the source code of this file.

Classes
class	mkldnn::handle_traits< T >
	A class that provides the destructor for an Intel(R) MKL-DNN C handle. More...
class	mkldnn::handle< T, traits >
	A class for wrapping an Intel(R) MKL-DNN handle. It is used as the base class for primitive (mkldnn_primitive_t), engine (mkldnn_engine_t), and stream (mkldnn_stream_t) handles. An object of the mkldnn::handle class can be passed by value. This class enables wrapping: More...
class	mkldnn::primitive
	Base class for all computational primitives. More...
struct	mkldnn::primitive::at
	A wrapper structure to specify a particular output of a primitive. More...
struct	mkldnn::error
	Intel(R) MKL-DNN exception class. More...
struct	mkldnn::post_ops
struct	mkldnn::primitive_attr
struct	mkldnn::engine
	An execution engine. More...
struct	mkldnn::memory
	Memory primitive that describes the data. More...
struct	mkldnn::memory::desc
	A memory descriptor. More...
struct	mkldnn::memory::primitive_desc
	A memory primitive descriptor. More...
struct	mkldnn::reorder
struct	mkldnn::reorder::primitive_desc
struct	mkldnn::view
struct	mkldnn::view::primitive_desc
struct	mkldnn::concat
struct	mkldnn::concat::primitive_desc
struct	mkldnn::sum
struct	mkldnn::sum::primitive_desc
struct	mkldnn::primitive_desc
	A base class for all primitive descriptors. More...
struct	mkldnn::convolution_forward
struct	mkldnn::convolution_forward::desc
struct	mkldnn::convolution_forward::primitive_desc
struct	mkldnn::convolution_backward_data
struct	mkldnn::convolution_backward_data::desc
struct	mkldnn::convolution_backward_data::primitive_desc
struct	mkldnn::convolution_backward_weights
struct	mkldnn::convolution_backward_weights::desc
struct	mkldnn::convolution_backward_weights::primitive_desc
struct	mkldnn::deconvolution_forward
struct	mkldnn::deconvolution_forward::desc
struct	mkldnn::deconvolution_forward::primitive_desc
struct	mkldnn::deconvolution_backward_data
struct	mkldnn::deconvolution_backward_data::desc
struct	mkldnn::deconvolution_backward_data::primitive_desc
struct	mkldnn::deconvolution_backward_weights
struct	mkldnn::deconvolution_backward_weights::desc
struct	mkldnn::deconvolution_backward_weights::primitive_desc
struct	mkldnn::lrn_forward
struct	mkldnn::lrn_forward::desc
struct	mkldnn::lrn_forward::primitive_desc
struct	mkldnn::lrn_backward
struct	mkldnn::lrn_backward::desc
struct	mkldnn::lrn_backward::primitive_desc
struct	mkldnn::pooling_forward
struct	mkldnn::pooling_forward::desc
struct	mkldnn::pooling_forward::primitive_desc
struct	mkldnn::pooling_backward
struct	mkldnn::pooling_backward::desc
struct	mkldnn::pooling_backward::primitive_desc
struct	mkldnn::eltwise_forward
struct	mkldnn::eltwise_forward::desc
struct	mkldnn::eltwise_forward::primitive_desc
struct	mkldnn::eltwise_backward
struct	mkldnn::eltwise_backward::desc
struct	mkldnn::eltwise_backward::primitive_desc
struct	mkldnn::softmax_forward
struct	mkldnn::softmax_forward::desc
struct	mkldnn::softmax_forward::primitive_desc
struct	mkldnn::softmax_backward
struct	mkldnn::softmax_backward::desc
struct	mkldnn::softmax_backward::primitive_desc
struct	mkldnn::batch_normalization_forward
struct	mkldnn::batch_normalization_forward::desc
struct	mkldnn::batch_normalization_forward::primitive_desc
struct	mkldnn::batch_normalization_backward
struct	mkldnn::batch_normalization_backward::desc
struct	mkldnn::batch_normalization_backward::primitive_desc
struct	mkldnn::inner_product_forward
struct	mkldnn::inner_product_forward::desc
struct	mkldnn::inner_product_forward::primitive_desc
struct	mkldnn::inner_product_backward_data
struct	mkldnn::inner_product_backward_data::desc
struct	mkldnn::inner_product_backward_data::primitive_desc
struct	mkldnn::inner_product_backward_weights
struct	mkldnn::inner_product_backward_weights::desc
struct	mkldnn::inner_product_backward_weights::primitive_desc
struct	mkldnn::rnn_cell
struct	mkldnn::rnn_cell::desc
struct	mkldnn::rnn_forward
struct	mkldnn::rnn_forward::desc
struct	mkldnn::rnn_forward::primitive_desc
struct	mkldnn::rnn_backward
struct	mkldnn::rnn_backward::desc
struct	mkldnn::rnn_backward::primitive_desc
struct	mkldnn::shuffle_forward
struct	mkldnn::shuffle_forward::desc
struct	mkldnn::shuffle_forward::primitive_desc
struct	mkldnn::shuffle_backward
struct	mkldnn::shuffle_backward::desc
struct	mkldnn::shuffle_backward::primitive_desc
struct	mkldnn::stream

Namespaces
	mkldnn

Macros
#define	REG_QUERY_MPD(name, what, idx)

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_primitive_kind_t	mkldnn::convert_to_c (primitive::kind akind)
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)
memory::desc	mkldnn::zero_md ()
memory	mkldnn::null_memory (engine eng)
void	mkldnn::check_num_parameters (const const_mkldnn_primitive_desc_t &aprimitive_desc, int n_inputs, int n_outputs, const std::string &prim_name)
bool	mkldnn::is_null_memory (const const_mkldnn_primitive_t &aprimitive)
bool	mkldnn::operator== (mkldnn_data_type_t a, memory::data_type b)
bool	mkldnn::operator!= (mkldnn_data_type_t a, memory::data_type b)
bool	mkldnn::operator== (memory::data_type a, mkldnn_data_type_t b)
bool	mkldnn::operator!= (memory::data_type a, mkldnn_data_type_t b)
bool	mkldnn::operator== (mkldnn_memory_format_t a, memory::format b)
bool	mkldnn::operator!= (mkldnn_memory_format_t a, memory::format b)
bool	mkldnn::operator== (memory::format a, mkldnn_memory_format_t b)
bool	mkldnn::operator!= (memory::format a, mkldnn_memory_format_t b)

Macro Definition Documentation

REG_QUERY_MPD

#define REG_QUERY_MPD	(		name,
			what,
			idx
	)

Value:

memory::primitive_desc name ## _primitive_desc() const \
    { return query_mpd(what ## _pd, idx); }