|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_desc_iterator_create (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) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_desc_iterator_next (mkldnn_primitive_desc_iterator_t iterator) |
|
mkldnn_primitive_desc_t MKLDNN_API | mkldnn_primitive_desc_iterator_fetch (const_mkldnn_primitive_desc_iterator_t iterator) |
|
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, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc) |
|
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) |
|
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) |
|
const mkldnn_memory_desc_t MKLDNN_API * | mkldnn_primitive_desc_query_md (const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index) |
|
int MKLDNN_API | mkldnn_primitive_desc_query_s32 (const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_create (mkldnn_primitive_t *primitive, const_mkldnn_primitive_desc_t primitive_desc) |
| Creates a primitive using a primitive_desc descriptor. More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_execute (const_mkldnn_primitive_t primitive, mkldnn_stream_t stream, int nargs, const mkldnn_exec_arg_t *args) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_get_primitive_desc (const_mkldnn_primitive_t primitive, const_mkldnn_primitive_desc_t *primitive_desc) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_destroy (mkldnn_primitive_t primitive) |
| Deletes a primitive . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_create (mkldnn_primitive_attr_t *attr) |
|
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_scratchpad_mode (const_mkldnn_primitive_attr_t attr, mkldnn_scratchpad_mode_t *mode) |
| Returns the scratchpad mode set in the attribute attr . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_set_scratchpad_mode (mkldnn_primitive_attr_t attr, mkldnn_scratchpad_mode_t mode) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_get_output_scales (const_mkldnn_primitive_attr_t attr, mkldnn_dim_t *count, int *mask, const float **scales) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_set_output_scales (mkldnn_primitive_attr_t attr, mkldnn_dim_t count, int mask, const float *scales) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_get_post_ops (const_mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t *post_ops) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_set_post_ops (mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t post_ops) |
|
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) |
|
mkldnn_status_t MKLDNN_API | mkldnn_post_ops_append_sum (mkldnn_post_ops_t post_ops, float scale) |
|
mkldnn_status_t MKLDNN_API | mkldnn_post_ops_get_params_sum (const_mkldnn_post_ops_t post_ops, int index, float *scale) |
|
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) |
|
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) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_desc_init_by_strides (mkldnn_memory_desc_t *memory_desc, int ndims, const mkldnn_dims_t dims, mkldnn_data_type_t data_type, const mkldnn_dims_t strides) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_desc_init_by_tag (mkldnn_memory_desc_t *memory_desc, int ndims, const mkldnn_dims_t dims, mkldnn_data_type_t data_type, mkldnn_format_tag_t tag) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_desc_init_submemory (mkldnn_memory_desc_t *memory_desc, const mkldnn_memory_desc_t *parent_memory_desc, const mkldnn_dims_t dims, const mkldnn_dims_t offsets) |
|
int MKLDNN_API | mkldnn_memory_desc_equal (const mkldnn_memory_desc_t *lhs, const mkldnn_memory_desc_t *rhs) |
|
size_t MKLDNN_API | mkldnn_memory_desc_get_size (const mkldnn_memory_desc_t *memory_desc) |
| Returns the size (in bytes) that is required for given memory_desc . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_create (mkldnn_memory_t *memory, const mkldnn_memory_desc_t *memory_desc, mkldnn_engine_t engine, void *handle) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_get_memory_desc (const_mkldnn_memory_t memory, const mkldnn_memory_desc_t **memory_desc) |
| Returns a memory_desc associated with memory . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_get_engine (const_mkldnn_memory_t memory, mkldnn_engine_t *engine) |
| Returns an engine associated with memory . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_map_data (const_mkldnn_memory_t memory, void **mapped_ptr) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_unmap_data (const_mkldnn_memory_t memory, void *mapped_ptr) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_get_data_handle (const_mkldnn_memory_t memory, void **handle) |
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_set_data_handle (mkldnn_memory_t memory, void *handle) |
| For a memory , sets the data handle . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_get_ocl_mem_object (const_mkldnn_memory_t memory, cl_mem *mem_object) |
| For a memory returns the OpenCL memory object associated with it. More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_set_ocl_mem_object (mkldnn_memory_t memory, cl_mem mem_object) |
| For a memory sets the OpenCL memory object associated with it. More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_memory_destroy (mkldnn_memory_t memory) |
| Deletes a memory . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_reorder_primitive_desc_create (mkldnn_primitive_desc_t *reorder_primitive_desc, const mkldnn_memory_desc_t *src_md, mkldnn_engine_t src_engine, const mkldnn_memory_desc_t *dst_md, mkldnn_engine_t dst_engine, const_mkldnn_primitive_attr_t attr) |
|
mkldnn_status_t MKLDNN_API | mkldnn_concat_primitive_desc_create (mkldnn_primitive_desc_t *concat_primitive_desc, const mkldnn_memory_desc_t *dst_md, int n, int concat_dimension, const mkldnn_memory_desc_t *src_mds, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine) |
|
mkldnn_status_t MKLDNN_API | mkldnn_sum_primitive_desc_create (mkldnn_primitive_desc_t *sum_primitive_desc, const mkldnn_memory_desc_t *dst_mds, int n, const float *scales, const mkldnn_memory_desc_t *src_mds, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine) |
|
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_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_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_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_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_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_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_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_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_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_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_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_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, mkldnn_dim_t group_size) |
|
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, mkldnn_dim_t group_size) |
|
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) |
|
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) |
|
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) |
|
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) |
|
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_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_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, mkldnn_dim_t local_size, float alpha, float beta, float k) |
|
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, mkldnn_dim_t local_size, float alpha, float beta, float k) |
|
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) |
|
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) |
|
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) |
|
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) |
|
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) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_set_rnn_data_qparams (mkldnn_primitive_attr_t attr, const float scale, const float shift) |
|
mkldnn_status_t MKLDNN_API | mkldnn_primitive_attr_set_rnn_weights_qparams (mkldnn_primitive_attr_t attr, mkldnn_dim_t count, int mask, const float *weights_scales) |
|
mkldnn_status_t MKLDNN_API | mkldnn_vanilla_rnn_forward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_alg_kind_t activation, 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, unsigned flags, float alpha, float beta) |
|
mkldnn_status_t MKLDNN_API | mkldnn_vanilla_rnn_backward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_alg_kind_t activation, 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_desc, const mkldnn_memory_desc_t *diff_dst_iter_desc, unsigned flags, float alpha, float beta) |
|
mkldnn_status_t MKLDNN_API | mkldnn_lstm_forward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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 *src_iter_c_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 *dst_iter_c_desc, unsigned flags) |
|
mkldnn_status_t MKLDNN_API | mkldnn_lstm_backward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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 *src_iter_c_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 *dst_iter_c_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_src_iter_c_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_desc, const mkldnn_memory_desc_t *diff_dst_iter_desc, const mkldnn_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) |
|
mkldnn_status_t MKLDNN_API | mkldnn_gru_forward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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, unsigned flags) |
|
mkldnn_status_t MKLDNN_API | mkldnn_gru_backward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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_desc, const mkldnn_memory_desc_t *diff_dst_iter_desc, unsigned flags) |
|
mkldnn_status_t MKLDNN_API | mkldnn_lbr_gru_forward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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, unsigned flags) |
|
mkldnn_status_t MKLDNN_API | mkldnn_lbr_gru_backward_desc_init (mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, 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_desc, const mkldnn_memory_desc_t *diff_dst_iter_desc, unsigned flags) |
|
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_create_ocl (mkldnn_engine_t *engine, mkldnn_engine_kind_t kind, cl_device_id device, cl_context context) |
|
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_get_ocl_context (mkldnn_engine_t engine, cl_context *context) |
| Returns an OpenCL context associated with an engine . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_engine_get_ocl_device (mkldnn_engine_t engine, cl_device_id *device) |
| Returns an OpenCL device associated with 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_engine_t engine, unsigned flags) |
| Creates an execution stream for engine and with flags . More...
|
|
mkldnn_status_t MKLDNN_API | mkldnn_stream_create_ocl (mkldnn_stream_t *stream, mkldnn_engine_t engine, cl_command_queue queue) |
|
mkldnn_status_t MKLDNN_API | mkldnn_stream_get_ocl_command_queue (mkldnn_stream_t stream, cl_command_queue *queue) |
|
mkldnn_status_t MKLDNN_API | mkldnn_stream_wait (mkldnn_stream_t stream) |
| Waits for all primitives in the execution stream to finish. 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) |
|
mkldnn_status_t MKLDNN_API | mkldnn_set_jit_dump (int enable) |
|
const mkldnn_version_t MKLDNN_API * | mkldnn_version () |
|
mkldnn_status_t MKLDNN_API | mkldnn_sgemm (char transa, char transb, mkldnn_dim_t M, mkldnn_dim_t N, mkldnn_dim_t K, float alpha, const float *A, mkldnn_dim_t lda, const float *B, mkldnn_dim_t ldb, float beta, float *C, mkldnn_dim_t ldc) |
|
mkldnn_status_t MKLDNN_API | mkldnn_gemm_u8s8s32 (char transa, char transb, char offsetc, mkldnn_dim_t M, mkldnn_dim_t N, mkldnn_dim_t K, float alpha, const uint8_t *A, mkldnn_dim_t lda, uint8_t ao, const int8_t *B, mkldnn_dim_t ldb, int8_t bo, float beta, int32_t *C, mkldnn_dim_t ldc, const int32_t *co) |
|
mkldnn_status_t MKLDNN_API | mkldnn_gemm_s8s8s32 (char transa, char transb, char offsetc, mkldnn_dim_t M, mkldnn_dim_t N, mkldnn_dim_t K, float alpha, const int8_t *A, mkldnn_dim_t lda, int8_t ao, const int8_t *B, mkldnn_dim_t ldb, int8_t bo, float beta, int32_t *C, mkldnn_dim_t ldc, const int32_t *co) |
|