mkldnn.h File Reference

Go to the source code of this file.

Functions
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_iterator_create (mkldnn_primitive_desc_iterator_t *iterator, const_mkldnn_op_desc_t op_desc, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
	Creates a primitive descriptor iterator for given op_desc, engine, and optionally a hint primitive descriptor from forward propagation (required for backward propagation). More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_iterator_create_v2 (mkldnn_primitive_desc_iterator_t *iterator, const_mkldnn_op_desc_t op_desc, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
	Creates a primitive descriptor iterator for given op_desc, attr, engine, and optionally a hint primitive descriptor from forward propagation (required for backward propagation). More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_iterator_next (mkldnn_primitive_desc_iterator_t iterator)
	Iterates over primitive descriptors. More...
mkldnn_primitive_desc_t MKLDNN_API	mkldnn_primitive_desc_iterator_fetch (const_mkldnn_primitive_desc_iterator_t iterator)
	Fetches the current primitive descriptor. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_iterator_destroy (mkldnn_primitive_desc_iterator_t iterator)
	Deletes a primitive descriptor iterator. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_create (mkldnn_primitive_desc_t *primitive_desc, const_mkldnn_op_desc_t op_desc, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
	Creates a primitive_desc using op_desc, engine, and optionally a hint primitive descriptor from forward propagation. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_create_v2 (mkldnn_primitive_desc_t *primitive_desc, const_mkldnn_op_desc_t op_desc, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
	Creates a primitive_desc using op_desc, attr, engine, and optionally a hint primitive descriptor from forward propagation. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_clone (mkldnn_primitive_desc_t *primitive_desc, const_mkldnn_primitive_desc_t existing_primitive_desc)
	Makes a copy of a primitive_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_get_attr (const_mkldnn_primitive_desc_t primitive_desc, const_mkldnn_primitive_attr_t *attr)
	Returns a constant reference to the attribute of a primitive_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_destroy (mkldnn_primitive_desc_t primitive_desc)
	Deletes a primitive_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_desc_query (const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index, void *result)
	Queries primitive descriptor. More...
const mkldnn_memory_desc_t MKLDNN_API *	mkldnn_primitive_desc_query_memory_d (const_mkldnn_primitive_desc_t primitive_desc)
	Queries primitive descriptor for memory descriptor. More...
const_mkldnn_primitive_desc_t MKLDNN_API	mkldnn_primitive_desc_query_pd (const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index)
	Queries primitive descriptor for primitive descriptor. More...
int MKLDNN_API	mkldnn_primitive_desc_query_s32 (const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index)
	Queries primitive descriptor for signed 32bit int. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_create (mkldnn_primitive_t *primitive, const_mkldnn_primitive_desc_t primitive_desc, const mkldnn_primitive_at_t *inputs, const_mkldnn_primitive_t *outputs)
	Creates a primitive using a primitive_desc descriptor and arrays of inputs and outputs. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_get_primitive_desc (const_mkldnn_primitive_t primitive, const_mkldnn_primitive_desc_t *primitive_desc)
	Retrieves a reference to the primitive_desc descriptor of given primitive. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_get_input_at (const_mkldnn_primitive_t primitive, size_t index, mkldnn_primitive_at_t *input)
	For a primitive, returns input at the index position. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_get_output (const_mkldnn_primitive_t primitive, size_t index, const_mkldnn_primitive_t *output)
	For a primitive, returns output at the index position. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_destroy (mkldnn_primitive_t primitive)
	Deletes a primitive. More...
mkldnn_primitive_at_t MKLDNN_API	mkldnn_primitive_at (const_mkldnn_primitive_t primitive, size_t output_index)
	Creates an mkldnn_primitive_at_t structure from a primitive and output_index. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_create (mkldnn_primitive_attr_t *attr)
	Creates an empty (default) attr attribute. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_clone (mkldnn_primitive_attr_t *attr, const_mkldnn_primitive_attr_t existing_attr)
	Makes a copy of an existing_attr. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_destroy (mkldnn_primitive_attr_t attr)
	Deletes an attr. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_get_int_output_round_mode (const_mkldnn_primitive_attr_t attr, mkldnn_round_mode_t *round_mode)
	Returns integer output rounding mode round_mode for a given attr, previously set by mkldnn_primitive_attr_set_int_output_round_mode. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_set_int_output_round_mode (mkldnn_primitive_attr_t attr, mkldnn_round_mode_t round_mode)
	Sets output rounding mode round_mode for integer operations for a given attr. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_get_output_scales (const_mkldnn_primitive_attr_t attr, int *count, int *mask, const float **scales)
	Returns count, correspondence scale mask, and a pointer to a constant floating point array of output scales for given attr, previously set by mkldnn_primitive_attr_set_output_scales. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_set_output_scales (mkldnn_primitive_attr_t attr, int count, int mask, const float *scales)
	Sets output scales for primitive operations. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_get_post_ops (const_mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t *post_ops)
	Returns post_ops for given attr. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_set_post_ops (mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t post_ops)
	Sets configured post_ops to an attribute attr for future use (when primitive descriptor is being created). More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_create (mkldnn_post_ops_t *post_ops)
	Creates an empty sequence of post operations post_ops. More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_destroy (mkldnn_post_ops_t post_ops)
	Deletes a post_ops sequence. More...
int MKLDNN_API	mkldnn_post_ops_len (const_mkldnn_post_ops_t post_ops)
	Returns the length of post operations for given post_ops. More...
mkldnn_primitive_kind_t MKLDNN_API	mkldnn_post_ops_get_kind (const_mkldnn_post_ops_t post_ops, int index)
	Returns the type of post operation with index index in given post_ops. More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_append_sum (mkldnn_post_ops_t post_ops, float scale)
	Appends accumulation (sum) post operation to the post_ops. More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_get_params_sum (const_mkldnn_post_ops_t post_ops, int index, float *scale)
	Gets the parameters of the accumulation (sum) post operation with index index in the sequence of post_ops. More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_append_eltwise (mkldnn_post_ops_t post_ops, float scale, mkldnn_alg_kind_t alg, float alpha, float beta)
	Appends eltwise post operation to the post_ops with given parameters kind, alpha, and beta (. More...
mkldnn_status_t MKLDNN_API	mkldnn_post_ops_get_params_eltwise (const_mkldnn_post_ops_t post_ops, int index, float *scale, mkldnn_alg_kind_t *alg, float *alpha, float *beta)
	Gets the eltwise parameters of the post operation with index index in the sequence of post_ops. More...
mkldnn_status_t MKLDNN_API	mkldnn_memory_desc_init (mkldnn_memory_desc_t *memory_desc, int ndims, const mkldnn_dims_t dims, mkldnn_data_type_t data_type, mkldnn_memory_format_t format)
	Initializes a memory_desc memory descriptor using ndims, dims, data_type, and data format. More...
mkldnn_status_t MKLDNN_API	mkldnn_memory_primitive_desc_create (mkldnn_primitive_desc_t *memory_primitive_desc, const mkldnn_memory_desc_t *memory_desc, mkldnn_engine_t engine)
	Creates a memory_primitive_desc memory primitive descriptor using memory_desc and engine. More...
mkldnn_status_t MKLDNN_API	mkldnn_view_primitive_desc_create (mkldnn_primitive_desc_t *view_primitive_desc, const_mkldnn_primitive_desc_t memory_primitive_desc, const mkldnn_dims_t dims, const mkldnn_dims_t offsets)
	Creates a view_primitive_desc for a given memory_primitive_desc, with dims sizes and offsets offsets. More...
int MKLDNN_API	mkldnn_memory_primitive_desc_equal (const_mkldnn_primitive_desc_t lhs, const_mkldnn_primitive_desc_t rhs)
	Compares two descriptors of memory primitives. More...
size_t MKLDNN_API	mkldnn_memory_primitive_desc_get_size (const_mkldnn_primitive_desc_t memory_primitive_desc)
	Returns the size (in bytes) that is required for given memory_primitive_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_memory_get_data_handle (const_mkldnn_primitive_t memory, void **handle)
	For a memory primitive, returns the data handle. More...
mkldnn_status_t MKLDNN_API	mkldnn_memory_set_data_handle (mkldnn_primitive_t memory, void *handle)
	For a memory primitive, sets the data handle. More...
mkldnn_status_t MKLDNN_API	mkldnn_reorder_primitive_desc_create (mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output)
	Initializes a reorder_primitive_desc using descriptors of input and output memory primitives. More...
mkldnn_status_t MKLDNN_API	mkldnn_reorder_primitive_desc_create_v2 (mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output, const_mkldnn_primitive_attr_t attr)
	Initializes a reorder_primitive_desc using an attr attribute and descriptors of input and output memory primitives. More...
mkldnn_status_t MKLDNN_API	mkldnn_concat_primitive_desc_create (mkldnn_primitive_desc_t *concat_primitive_desc, const mkldnn_memory_desc_t *output_desc, int n, int concat_dimension, const_mkldnn_primitive_desc_t *input_pds)
	Creates out-of-place concat_primitive_desc for concatenation of n inputs by concat_dimension with resulting output_desc memory descriptor. More...
mkldnn_status_t MKLDNN_API	mkldnn_sum_primitive_desc_create (mkldnn_primitive_desc_t *sum_primitive_desc, const mkldnn_memory_desc_t *output_desc, int n, const float *scales, const_mkldnn_primitive_desc_t *input_pds)
	Creates out-of-place sum_primitive_desc for sum of n inputs multiplied by scale with resulting output_desc memory descriptor. More...
mkldnn_status_t MKLDNN_API	mkldnn_convolution_forward_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a convolution descriptor conv_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_convolution_forward_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a dilated convolution descriptor conv_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptors, strides, dilates, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_convolution_backward_data_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a convolution descriptor conv_desc for backward propagation with respect to data using alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_convolution_backward_data_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a dilated convolution descriptor conv_desc for backward propagation with respect to data using alg_kind, memory descriptors, strides, dilates padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_convolution_backward_weights_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a convolution descriptor conv_desc for backward propagation with respect to weights using alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_convolution_backward_weights_desc_init (mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a convolution descriptor conv_desc for backward propagation with respect to weights using alg_kind, memory descriptors, strides, dilates padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_deconvolution_forward_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a deconvolution descriptor deconv_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_deconvolution_forward_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a dilated deconvolution descriptor deconv_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptors, strides, dilates, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_deconvolution_backward_data_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a deconvolution descriptor conv_desc for backward propagation with respect to data using alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_deconvolution_backward_data_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a dilated deconvolution descriptor conv_desc for backward propagation with respect to data using alg_kind, memory descriptors, strides, dilates, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_deconvolution_backward_weights_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a deconvolution descriptor conv_desc for backward propagation with respect to weights using alg_kind, memory descriptors, strides, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_dilated_deconvolution_backward_weights_desc_init (mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t 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 mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a dilated deconvolution descriptor conv_desc for backward propagation with respect to weights using alg_kind, memory descriptors, strides, dilates, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_shuffle_forward_desc_init (mkldnn_shuffle_desc_t *shuffle_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, int axis, int group_size)
	Initializes a shuffle_desc for forward propagation using prop_kind, memory descriptor data_desc, axis, and group_size. More...
mkldnn_status_t MKLDNN_API	mkldnn_shuffle_backward_desc_init (mkldnn_shuffle_desc_t *shuffle_desc, const mkldnn_memory_desc_t *diff_data_desc, int axis, int group_size)
	Initializes a shuffle_desc for backward propagation using memory descriptor diff_data_desc, axis, and group_size. More...
mkldnn_status_t MKLDNN_API	mkldnn_eltwise_forward_desc_init (mkldnn_eltwise_desc_t *eltwise_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc, float alpha, float beta)
	Initializes an eltwise_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind algorithm, memory descriptor data_desc, alpha, and beta parameters. More...
mkldnn_status_t MKLDNN_API	mkldnn_eltwise_backward_desc_init (mkldnn_eltwise_desc_t *eltwise_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, float alpha, float beta)
	Initializes an eltwise_desc for backward propagation using alg_kind algorithm memory descriptors diff_data_desc and data_desc, and the alpha and beta parameters. More...
mkldnn_status_t MKLDNN_API	mkldnn_softmax_forward_desc_init (mkldnn_softmax_desc_t *softmax_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, int softmax_axis)
	Initializes a softmax_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference) and memory descriptor data_desc. More...
mkldnn_status_t MKLDNN_API	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)
	Initializes a softmax_desc for backward propagation using memory descriptors diff_desc and data_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_pooling_forward_desc_init (mkldnn_pooling_desc_t *pool_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a pooling descriptor pool_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptors, and pooling parameters in the spatial domain: strides, kernel sizes, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_pooling_backward_desc_init (mkldnn_pooling_desc_t *pool_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
	Initializes a pooling descriptor pool_desc for backward propagation using alg_kind, memory descriptors, and pooling parameters in the spatial domain: strides, kernel sizes, padding_l, padding_r, and padding_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_lrn_forward_desc_init (mkldnn_lrn_desc_t *lrn_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc, int local_size, float alpha, float beta, float k)
	Initializes an lrn_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), alg_kind, memory descriptor data_desc, and regularization parameters local_size, alpha, beta, and k. More...
mkldnn_status_t MKLDNN_API	mkldnn_lrn_backward_desc_init (mkldnn_lrn_desc_t *lrn_desc, mkldnn_alg_kind_t 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)
	Initializes an lrn_desc for backward propagation using alg_kind, memory descriptors data_desc and diff_data_desc, and regularization parameters local_size, alpha, beta, and k. More...
mkldnn_status_t MKLDNN_API	mkldnn_batch_normalization_forward_desc_init (mkldnn_batch_normalization_desc_t *bnrm_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, float epsilon, unsigned flags)
	Initializes a batch normalization descriptor bnrm_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference), memory descriptor data_desc, normalization parameter epsilon, and flags set using bit flags of type mkldnn_batch_normalization_desc_t. More...
mkldnn_status_t MKLDNN_API	mkldnn_batch_normalization_backward_desc_init (mkldnn_batch_normalization_desc_t *bnrm_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, float epsilon, unsigned flags)
	Initializes a batch normalization descriptor bnrm_desc for backward propagation with respect to data and scale-shift parameters using memory descriptors data_desc and diff_data_desc, normalization parameter epsilon, and flags set using bit flags of type mkldnn_batch_normalization_desc_t. More...
mkldnn_status_t MKLDNN_API	mkldnn_inner_product_forward_desc_init (mkldnn_inner_product_desc_t *ip_desc, mkldnn_prop_kind_t 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)
	Initializes an inner product descriptor ip_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training and mkldnn_forward_inference) and memory descriptors. More...
mkldnn_status_t MKLDNN_API	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)
	Initializes an inner product descriptor ip_desc for backward propagation with respect to data using memory descriptors. More...
mkldnn_status_t MKLDNN_API	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)
	Initializes an inner product descriptor ip_desc for backward propagation with respect to weights using memory descriptors. More...
mkldnn_status_t MKLDNN_API	mkldnn_rnn_cell_desc_init (mkldnn_rnn_cell_desc_t *rnn_cell_desc, mkldnn_alg_kind_t kind, mkldnn_alg_kind_t f, unsigned int flags, float alpha, float clipping)
	Initializes a recurrent cell descriptor rnn_cell_desc using rnn_cell_desc, kind (possible values are mkldnn_vanilla_rnn, mkldnn_vanilla_lstm, mkldnn_vanilla_gru, and mkldnn_gru_linear_before_reset), f (possible values are mkldnn_eltwise_relu and mkldnn_eltwise_tanh), flags, alpha, and clipping. More...
int MKLDNN_API	mkldnn_rnn_cell_get_gates_count (const mkldnn_rnn_cell_desc_t *rnn_cell_desc)
	Returns the number of gates of a particular rnn_cell_desc. More...
int MKLDNN_API	mkldnn_rnn_cell_get_states_count (const mkldnn_rnn_cell_desc_t *rnn_cell_desc)
	Returns the number of states of a particular rnn_cell_desc. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_set_rnn_data_qparams (mkldnn_primitive_attr_t attr, const float scale, const float shift)
	Sets quantization scale and shift for RNN data tensors. More...
mkldnn_status_t MKLDNN_API	mkldnn_primitive_attr_set_rnn_weights_qparams (mkldnn_primitive_attr_t attr, int count, int mask, const float *weights_scales)
	Sets quantization scales weights_scales for RNN weights tensors. More...
mkldnn_status_t MKLDNN_API	mkldnn_rnn_forward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_rnn_cell_desc_t *rnn_cell_desc, const mkldnn_rnn_direction_t 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)
	Initializes a rnn descriptor rnn_desc for forward propagation using prop_kind, rnn_cell_desc, direction, and memory descriptors. More...
mkldnn_status_t MKLDNN_API	mkldnn_rnn_backward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_rnn_cell_desc_t *rnn_cell_desc, const mkldnn_rnn_direction_t 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)
	Initializes a rnn descriptor rnn_desc for backward propagation using prop_kind, rnn_cell_desc, direction, and memory descriptors. More...
size_t MKLDNN_API	mkldnn_engine_get_count (mkldnn_engine_kind_t kind)
	Returns the number of engines of a particular kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_engine_create (mkldnn_engine_t *engine, mkldnn_engine_kind_t kind, size_t index)
	Creates an engine of particular kind and index. More...
mkldnn_status_t MKLDNN_API	mkldnn_engine_get_kind (mkldnn_engine_t engine, mkldnn_engine_kind_t *kind)
	Returns the kind of an engine. More...
mkldnn_status_t MKLDNN_API	mkldnn_engine_destroy (mkldnn_engine_t engine)
	Destroys an engine. More...
mkldnn_status_t MKLDNN_API	mkldnn_stream_create (mkldnn_stream_t *stream, mkldnn_stream_kind_t stream_kind)
	Creates an execution stream of stream_kind. More...
mkldnn_status_t MKLDNN_API	mkldnn_stream_submit (mkldnn_stream_t stream, size_t n, mkldnn_primitive_t primitives[], mkldnn_primitive_t *error_primitive)
	Submits primitives to an execution stream. More...
mkldnn_status_t MKLDNN_API	mkldnn_stream_wait (mkldnn_stream_t stream, int block, mkldnn_primitive_t *error_primitive)
	Waits for all primitives in the execution stream to finish. More...
mkldnn_status_t MKLDNN_API	mkldnn_stream_rerun (mkldnn_stream_t stream, mkldnn_primitive_t *error_primitive)
	Reruns all the primitives within the stream. More...
mkldnn_status_t MKLDNN_API	mkldnn_stream_destroy (mkldnn_stream_t stream)
	Destroys an execution stream. More...
mkldnn_status_t MKLDNN_API	mkldnn_set_verbose (int level)
	Sets verbosity level (print information to stdout). More...
mkldnn_status_t MKLDNN_API	mkldnn_set_jit_dump (int dump)
	Sets jit dump control. More...
const mkldnn_version_t MKLDNN_API *	mkldnn_version ()
	Gets library version information. More...
mkldnn_status_t MKLDNN_API	mkldnn_sgemm (const char *transa, const char *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)
	SGEMM performs a matrix-matrix multiplication operation defined as. More...
mkldnn_status_t MKLDNN_API	mkldnn_gemm_s8u8s32 (const char *transa, const char *transb, const char *offsetc, const int *M, const int *N, const int *K, const float *alpha, const int8_t *A, const int *lda, const int8_t *ao, const uint8_t *B, const int *ldb, const int8_t *bo, const float *beta, int32_t *c, const int *ldc, const int32_t *co)
	gemm_s8u8s32 and gemm_s8s8s32 perform a matrix-matrix multiplication operation and add the result to a scalar-matrix product. More...
mkldnn_status_t MKLDNN_API	mkldnn_gemm_s8s8s32 (const char *transa, const char *transb, const char *offsetc, const int *M, const int *N, const int *K, const float *alpha, const int8_t *A, const int *lda, const int8_t *ao, const int8_t *B, const int *ldb, const int8_t *bo, const float *beta, int32_t *c, const int *ldc, const int32_t *co)
mkldnn_status_t MKLDNN_API	mkldnn_gemm_bf16bf16f32 (const char *transa, const char *transb, const int *M, const int *N, const int *K, const float *alpha, const mkldnn_bfloat16_t *A, const int *lda, const mkldnn_bfloat16_t *B, const int *ldb, const float *beta, float *c, const int *ldc)
	gemm_bf16bf16f32 performs a matrix-matrix multiplication operation defined as More...