Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  0.19.0
Performance library for Deep Learning
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Functions
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)