#include "dnnl_config.h"
#include "dnnl_types.h"
#include "dnnl_version.h"

Include dependency graph for dnnl.h:

This graph shows which files directly or indirectly include this file:

Functions
dnnl_status_t DNNL_API	dnnl_primitive_desc_iterator_create (dnnl_primitive_desc_iterator_t *iterator, const_dnnl_op_desc_t op_desc, const_dnnl_primitive_attr_t attr, dnnl_engine_t engine, const_dnnl_primitive_desc_t hint_forward_primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_desc_iterator_next (dnnl_primitive_desc_iterator_t iterator)

dnnl_primitive_desc_t DNNL_API	dnnl_primitive_desc_iterator_fetch (const_dnnl_primitive_desc_iterator_t iterator)

dnnl_status_t DNNL_API	dnnl_primitive_desc_iterator_destroy (dnnl_primitive_desc_iterator_t iterator)

dnnl_status_t DNNL_API	dnnl_primitive_desc_create (dnnl_primitive_desc_t *primitive_desc, const_dnnl_op_desc_t op_desc, const_dnnl_primitive_attr_t attr, dnnl_engine_t engine, const_dnnl_primitive_desc_t hint_forward_primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_desc_clone (dnnl_primitive_desc_t *primitive_desc, const_dnnl_primitive_desc_t existing_primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_desc_get_attr (const_dnnl_primitive_desc_t primitive_desc, const_dnnl_primitive_attr_t *attr)

dnnl_status_t DNNL_API	dnnl_primitive_desc_destroy (dnnl_primitive_desc_t primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_desc_query (const_dnnl_primitive_desc_t primitive_desc, dnnl_query_t what, int index, void *result)

const dnnl_memory_desc_t DNNL_API *	dnnl_primitive_desc_query_md (const_dnnl_primitive_desc_t primitive_desc, dnnl_query_t what, int index)

int DNNL_API	dnnl_primitive_desc_query_s32 (const_dnnl_primitive_desc_t primitive_desc, dnnl_query_t what, int index)

dnnl_status_t DNNL_API	dnnl_primitive_create (dnnl_primitive_t *primitive, const_dnnl_primitive_desc_t primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_execute (const_dnnl_primitive_t primitive, dnnl_stream_t stream, int nargs, const dnnl_exec_arg_t *args)

dnnl_status_t DNNL_API	dnnl_primitive_get_primitive_desc (const_dnnl_primitive_t primitive, const_dnnl_primitive_desc_t *primitive_desc)

dnnl_status_t DNNL_API	dnnl_primitive_destroy (dnnl_primitive_t primitive)

dnnl_status_t DNNL_API	dnnl_primitive_attr_create (dnnl_primitive_attr_t *attr)

dnnl_status_t DNNL_API	dnnl_primitive_attr_clone (dnnl_primitive_attr_t *attr, const_dnnl_primitive_attr_t existing_attr)

dnnl_status_t DNNL_API	dnnl_primitive_attr_destroy (dnnl_primitive_attr_t attr)

dnnl_status_t DNNL_API	dnnl_primitive_attr_get_scratchpad_mode (const_dnnl_primitive_attr_t attr, dnnl_scratchpad_mode_t *mode)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_scratchpad_mode (dnnl_primitive_attr_t attr, dnnl_scratchpad_mode_t mode)

dnnl_status_t DNNL_API	dnnl_primitive_attr_get_output_scales (const_dnnl_primitive_attr_t attr, dnnl_dim_t count, int mask, const float **scales)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_output_scales (dnnl_primitive_attr_t attr, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_primitive_attr_get_scales (dnnl_primitive_attr_t attr, int arg, dnnl_dim_t count, int mask, const float **scales)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_scales (dnnl_primitive_attr_t attr, int arg, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_primitive_attr_get_zero_points (const_dnnl_primitive_attr_t attr, int arg, dnnl_dim_t count, int mask, const int32_t **zero_points)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_zero_points (dnnl_primitive_attr_t attr, int arg, dnnl_dim_t count, int mask, const int32_t *zero_points)

dnnl_status_t DNNL_API	dnnl_primitive_attr_get_post_ops (const_dnnl_primitive_attr_t attr, const_dnnl_post_ops_t *post_ops)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_post_ops (dnnl_primitive_attr_t attr, const_dnnl_post_ops_t post_ops)

dnnl_status_t DNNL_API	dnnl_post_ops_create (dnnl_post_ops_t *post_ops)

dnnl_status_t DNNL_API	dnnl_post_ops_destroy (dnnl_post_ops_t post_ops)

int DNNL_API	dnnl_post_ops_len (const_dnnl_post_ops_t post_ops)

dnnl_primitive_kind_t DNNL_API	dnnl_post_ops_get_kind (const_dnnl_post_ops_t post_ops, int index)

dnnl_status_t DNNL_API	dnnl_post_ops_append_sum (dnnl_post_ops_t post_ops, float scale)

dnnl_status_t DNNL_API	dnnl_post_ops_append_sum_v2 (dnnl_post_ops_t post_ops, float scale, dnnl_data_type_t data_type)

dnnl_status_t DNNL_API	dnnl_post_ops_get_params_sum (const_dnnl_post_ops_t post_ops, int index, float *scale)

dnnl_status_t DNNL_API	dnnl_post_ops_get_params_sum_v2 (const_dnnl_post_ops_t post_ops, int index, float scale, dnnl_data_type_t data_type)

dnnl_status_t DNNL_API	dnnl_post_ops_append_eltwise (dnnl_post_ops_t post_ops, float scale, dnnl_alg_kind_t alg_kind, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_post_ops_get_params_eltwise (const_dnnl_post_ops_t post_ops, int index, float scale, dnnl_alg_kind_t alg_kind, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_post_ops_append_dw_k3s1p1 (dnnl_post_ops_t post_ops, dnnl_data_type_t weights_data_type, dnnl_data_type_t bias_data_type, dnnl_data_type_t dst_data_type, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_post_ops_get_params_dw_k3s1p1 (const_dnnl_post_ops_t post_ops, int index, dnnl_data_type_t weights_data_type, dnnl_data_type_t bias_data_type, dnnl_data_type_t dst_data_type, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_post_ops_append_dw_k3s2p1 (dnnl_post_ops_t post_ops, dnnl_data_type_t weights_data_type, dnnl_data_type_t bias_data_type, dnnl_data_type_t dst_data_type, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_post_ops_get_params_dw_k3s2p1 (const_dnnl_post_ops_t post_ops, int index, dnnl_data_type_t weights_data_type, dnnl_data_type_t bias_data_type, dnnl_data_type_t dst_data_type, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_memory_desc_init_by_strides (dnnl_memory_desc_t *memory_desc, int ndims, const dnnl_dims_t dims, dnnl_data_type_t data_type, const dnnl_dims_t strides)

dnnl_status_t DNNL_API	dnnl_memory_desc_init_by_tag (dnnl_memory_desc_t *memory_desc, int ndims, const dnnl_dims_t dims, dnnl_data_type_t data_type, dnnl_format_tag_t tag)

dnnl_status_t DNNL_API	dnnl_memory_desc_init_submemory (dnnl_memory_desc_t memory_desc, const dnnl_memory_desc_t parent_memory_desc, const dnnl_dims_t dims, const dnnl_dims_t offsets)

dnnl_status_t DNNL_API	dnnl_memory_desc_reshape (dnnl_memory_desc_t out_memory_desc, const dnnl_memory_desc_t in_memory_desc, int ndims, const dnnl_dims_t dims)

dnnl_status_t DNNL_API	dnnl_memory_desc_permute_axes (dnnl_memory_desc_t out_memory_desc, const dnnl_memory_desc_t in_memory_desc, const int *permutation)

int DNNL_API	dnnl_memory_desc_equal (const dnnl_memory_desc_t lhs, const dnnl_memory_desc_t rhs)

size_t DNNL_API	dnnl_memory_desc_get_size (const dnnl_memory_desc_t *memory_desc)

dnnl_status_t DNNL_API	dnnl_memory_create (dnnl_memory_t memory, const dnnl_memory_desc_t memory_desc, dnnl_engine_t engine, void *handle)

dnnl_status_t DNNL_API	dnnl_memory_get_memory_desc (const_dnnl_memory_t memory, const dnnl_memory_desc_t **memory_desc)

dnnl_status_t DNNL_API	dnnl_memory_get_engine (const_dnnl_memory_t memory, dnnl_engine_t *engine)

dnnl_status_t DNNL_API	dnnl_memory_map_data (const_dnnl_memory_t memory, void **mapped_ptr)

dnnl_status_t DNNL_API	dnnl_memory_unmap_data (const_dnnl_memory_t memory, void *mapped_ptr)

dnnl_status_t DNNL_API	dnnl_memory_get_data_handle (const_dnnl_memory_t memory, void **handle)

dnnl_status_t DNNL_API	dnnl_memory_set_data_handle (dnnl_memory_t memory, void *handle)

dnnl_status_t DNNL_API	dnnl_memory_set_data_handle_v2 (dnnl_memory_t memory, void *handle, dnnl_stream_t stream)

dnnl_status_t DNNL_API	dnnl_memory_destroy (dnnl_memory_t memory)

dnnl_status_t DNNL_API	dnnl_reorder_primitive_desc_create (dnnl_primitive_desc_t reorder_primitive_desc, const dnnl_memory_desc_t src_desc, dnnl_engine_t src_engine, const dnnl_memory_desc_t *dst_desc, dnnl_engine_t dst_engine, const_dnnl_primitive_attr_t attr)

dnnl_status_t DNNL_API	dnnl_concat_primitive_desc_create (dnnl_primitive_desc_t concat_primitive_desc, const dnnl_memory_desc_t dst_desc, int n, int concat_dimension, const dnnl_memory_desc_t *src_descs, const_dnnl_primitive_attr_t attr, dnnl_engine_t engine)

dnnl_status_t DNNL_API	dnnl_sum_primitive_desc_create (dnnl_primitive_desc_t sum_primitive_desc, const dnnl_memory_desc_t dst_desc, int n, const float scales, const dnnl_memory_desc_t src_descs, const_dnnl_primitive_attr_t attr, dnnl_engine_t engine)

dnnl_status_t DNNL_API	dnnl_binary_desc_init (dnnl_binary_desc_t binary_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src0_desc, const dnnl_memory_desc_t src1_desc, const dnnl_memory_desc_t dst_desc)

dnnl_status_t DNNL_API	dnnl_convolution_forward_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_convolution_forward_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_convolution_backward_data_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_convolution_backward_data_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_convolution_backward_weights_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t diff_weights_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_convolution_backward_weights_desc_init (dnnl_convolution_desc_t conv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t diff_weights_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_deconvolution_forward_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_deconvolution_forward_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_deconvolution_backward_data_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_deconvolution_backward_data_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_deconvolution_backward_weights_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t diff_weights_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_dilated_deconvolution_backward_weights_desc_init (dnnl_deconvolution_desc_t deconv_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t diff_weights_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t dilates, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_shuffle_forward_desc_init (dnnl_shuffle_desc_t shuffle_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, int axis, dnnl_dim_t group_size)

dnnl_status_t DNNL_API	dnnl_shuffle_backward_desc_init (dnnl_shuffle_desc_t shuffle_desc, const dnnl_memory_desc_t diff_data_desc, int axis, dnnl_dim_t group_size)

dnnl_status_t DNNL_API	dnnl_eltwise_forward_desc_init (dnnl_eltwise_desc_t eltwise_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t data_desc, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_eltwise_backward_desc_init (dnnl_eltwise_desc_t eltwise_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_softmax_forward_desc_init (dnnl_softmax_desc_t softmax_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, int softmax_axis)

dnnl_status_t DNNL_API	dnnl_softmax_backward_desc_init (dnnl_softmax_desc_t softmax_desc, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, int softmax_axis)

dnnl_status_t DNNL_API	dnnl_logsoftmax_forward_desc_init (dnnl_logsoftmax_desc_t logsoftmax_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, int logsoftmax_axis)

dnnl_status_t DNNL_API	dnnl_logsoftmax_backward_desc_init (dnnl_logsoftmax_desc_t logsoftmax_desc, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, int logsoftmax_axis)

dnnl_status_t DNNL_API	dnnl_pooling_forward_desc_init (dnnl_pooling_desc_t pool_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t *dst_desc, const dnnl_dims_t strides, const dnnl_dims_t kernel, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_pooling_backward_desc_init (dnnl_pooling_desc_t pool_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t *diff_dst_desc, const dnnl_dims_t strides, const dnnl_dims_t kernel, const dnnl_dims_t padding_l, const dnnl_dims_t padding_r)

dnnl_status_t DNNL_API	dnnl_lrn_forward_desc_init (dnnl_lrn_desc_t lrn_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t data_desc, dnnl_dim_t local_size, float alpha, float beta, float k)

dnnl_status_t DNNL_API	dnnl_lrn_backward_desc_init (dnnl_lrn_desc_t lrn_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, dnnl_dim_t local_size, float alpha, float beta, float k)

dnnl_status_t DNNL_API	dnnl_batch_normalization_forward_desc_init (dnnl_batch_normalization_desc_t bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, float epsilon, unsigned flags)

dnnl_status_t DNNL_API	dnnl_batch_normalization_backward_desc_init (dnnl_batch_normalization_desc_t bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, float epsilon, unsigned flags)

dnnl_status_t DNNL_API	dnnl_layer_normalization_forward_desc_init (dnnl_layer_normalization_desc_t lnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, const dnnl_memory_desc_t *stat_desc, float epsilon, unsigned flags)

dnnl_status_t DNNL_API	dnnl_layer_normalization_backward_desc_init (dnnl_layer_normalization_desc_t lnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t data_desc, const dnnl_memory_desc_t stat_desc, float epsilon, unsigned flags)

dnnl_status_t DNNL_API	dnnl_inner_product_forward_desc_init (dnnl_inner_product_desc_t ip_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc)

dnnl_status_t DNNL_API	dnnl_inner_product_backward_data_desc_init (dnnl_inner_product_desc_t ip_desc, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t diff_dst_desc)

dnnl_status_t DNNL_API	dnnl_inner_product_backward_weights_desc_init (dnnl_inner_product_desc_t ip_desc, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t diff_weights_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t *diff_dst_desc)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_rnn_data_qparams (dnnl_primitive_attr_t attr, const float scale, const float shift)

dnnl_status_t DNNL_API	dnnl_primitive_attr_set_rnn_weights_qparams (dnnl_primitive_attr_t attr, dnnl_dim_t count, int mask, const float *scales)

dnnl_status_t DNNL_API	dnnl_vanilla_rnn_forward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, const dnnl_alg_kind_t activation, const dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, unsigned flags, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_vanilla_rnn_backward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, const dnnl_alg_kind_t activation, const dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t *diff_dst_iter_desc, unsigned flags, float alpha, float beta)

dnnl_status_t DNNL_API	dnnl_lstm_forward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lstm_forward_desc_init_v2 (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t weights_peephole_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t *dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lstm_forward_desc_init_v3 (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t weights_peephole_desc, const dnnl_memory_desc_t weights_projection_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lstm_backward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t dst_iter_c_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_src_iter_c_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t diff_dst_iter_desc, const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lstm_backward_desc_init_v2 (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t weights_peephole_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t dst_iter_c_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_src_iter_c_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_weights_peephole_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t diff_dst_iter_desc, const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lstm_backward_desc_init_v3 (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t src_iter_c_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t weights_peephole_desc, const dnnl_memory_desc_t weights_projection_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t dst_iter_c_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_src_iter_c_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_weights_peephole_desc, const dnnl_memory_desc_t diff_weights_projection_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t diff_dst_iter_desc, const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_gru_forward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_gru_backward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t *diff_dst_iter_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lbr_gru_forward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_lbr_gru_backward_desc_init (dnnl_rnn_desc_t rnn_desc, dnnl_prop_kind_t prop_kind, dnnl_rnn_direction_t direction, const dnnl_memory_desc_t src_layer_desc, const dnnl_memory_desc_t src_iter_desc, const dnnl_memory_desc_t weights_layer_desc, const dnnl_memory_desc_t weights_iter_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t dst_layer_desc, const dnnl_memory_desc_t dst_iter_desc, const dnnl_memory_desc_t diff_src_layer_desc, const dnnl_memory_desc_t diff_src_iter_desc, const dnnl_memory_desc_t diff_weights_layer_desc, const dnnl_memory_desc_t diff_weights_iter_desc, const dnnl_memory_desc_t diff_bias_desc, const dnnl_memory_desc_t diff_dst_layer_desc, const dnnl_memory_desc_t *diff_dst_iter_desc, unsigned flags)

dnnl_status_t DNNL_API	dnnl_matmul_desc_init (dnnl_matmul_desc_t matmul_desc, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t weights_desc, const dnnl_memory_desc_t bias_desc, const dnnl_memory_desc_t *dst_desc)

dnnl_status_t DNNL_API	dnnl_resampling_forward_desc_init (dnnl_resampling_desc_t resampling_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const float factors, const dnnl_memory_desc_t src_desc, const dnnl_memory_desc_t dst_desc)

dnnl_status_t DNNL_API	dnnl_resampling_backward_desc_init (dnnl_resampling_desc_t resampling_desc, dnnl_alg_kind_t alg_kind, const float factors, const dnnl_memory_desc_t diff_src_desc, const dnnl_memory_desc_t diff_dst_desc)

size_t DNNL_API	dnnl_engine_get_count (dnnl_engine_kind_t kind)

dnnl_status_t DNNL_API	dnnl_engine_create (dnnl_engine_t *engine, dnnl_engine_kind_t kind, size_t index)

dnnl_status_t DNNL_API	dnnl_engine_get_kind (dnnl_engine_t engine, dnnl_engine_kind_t *kind)

dnnl_status_t DNNL_API	dnnl_engine_destroy (dnnl_engine_t engine)

dnnl_status_t DNNL_API	dnnl_stream_attr_create (dnnl_stream_attr_t *attr, dnnl_engine_kind_t kind)

dnnl_status_t DNNL_API	dnnl_stream_attr_destroy (dnnl_stream_attr_t attr)

dnnl_status_t DNNL_API	dnnl_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags)

dnnl_status_t DNNL_API	dnnl_stream_create_v2 (dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags, const_dnnl_stream_attr_t attr)

dnnl_status_t DNNL_API	dnnl_stream_wait (dnnl_stream_t stream)

dnnl_status_t DNNL_API	dnnl_stream_destroy (dnnl_stream_t stream)

dnnl_status_t DNNL_API	dnnl_get_primitive_cache_capacity (int *capacity)

dnnl_status_t DNNL_API	dnnl_set_primitive_cache_capacity (int capacity)

dnnl_status_t DNNL_API	dnnl_set_verbose (int level)

dnnl_status_t DNNL_API	dnnl_set_jit_dump (int enable)

const dnnl_version_t DNNL_API *	dnnl_version ()

dnnl_status_t DNNL_API	dnnl_set_jit_profiling_flags (unsigned flags)

dnnl_status_t DNNL_API	dnnl_set_jit_profiling_jitdumpdir (const char *dir)

dnnl_status_t DNNL_API	dnnl_set_max_cpu_isa (dnnl_cpu_isa_t isa)

dnnl_cpu_isa_t DNNL_API	dnnl_get_effective_cpu_isa (void)

dnnl_status_t DNNL_API	dnnl_sgemm (char transa, char transb, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const float A, dnnl_dim_t lda, const float B, dnnl_dim_t ldb, float beta, float *C, dnnl_dim_t ldc)

dnnl_status_t DNNL_API	dnnl_gemm_u8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const uint8_t A, dnnl_dim_t lda, uint8_t ao, const int8_t B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t C, dnnl_dim_t ldc, const int32_t co)

dnnl_status_t DNNL_API	dnnl_gemm_s8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const int8_t A, dnnl_dim_t lda, int8_t ao, const int8_t B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t C, dnnl_dim_t ldc, const int32_t co)

Detailed Description

C API

Functions

Detailed Description