mkl-dnn/html/dnnl_8h_source.html

/*******************************************************************************

* Copyright 2016-2020 Intel Corporation

*

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

*     http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*******************************************************************************/


#ifndef DNNL_H

#define DNNL_H


#include "dnnl_config.h"

#include "dnnl_types.h"

#include "dnnl_version.h"


#if DNNL_GPU_RUNTIME == DNNL_RUNTIME_OCL


// Set target version for OpenCL explicitly to suppress a compiler warning.

#ifndef CL_TARGET_OPENCL_VERSION

#define CL_TARGET_OPENCL_VERSION 120

#endif


#include <CL/cl.h>

#endif


#ifdef __cplusplus

extern "C" {

#endif


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_get_params_sum(

        const_dnnl_post_ops_t post_ops, int index, float *scale);


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);


#if DNNL_GPU_RUNTIME == DNNL_RUNTIME_OCL

dnnl_status_t DNNL_API dnnl_memory_get_ocl_mem_object(

        const_dnnl_memory_t memory, cl_mem *mem_object);


dnnl_status_t DNNL_API dnnl_memory_set_ocl_mem_object(

        dnnl_memory_t memory, cl_mem mem_object);

#endif


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);


#if DNNL_GPU_RUNTIME == DNNL_RUNTIME_OCL

dnnl_status_t DNNL_API dnnl_engine_create_ocl(dnnl_engine_t *engine,

        dnnl_engine_kind_t kind, cl_device_id device, cl_context context);

#endif


dnnl_status_t DNNL_API dnnl_engine_get_kind(

        dnnl_engine_t engine, dnnl_engine_kind_t *kind);


#if DNNL_GPU_RUNTIME == DNNL_RUNTIME_OCL

dnnl_status_t DNNL_API dnnl_engine_get_ocl_context(

        dnnl_engine_t engine, cl_context *context);


dnnl_status_t DNNL_API dnnl_engine_get_ocl_device(

        dnnl_engine_t engine, cl_device_id *device);

#endif


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);


#if DNNL_CPU_THREADING_RUNTIME == DNNL_RUNTIME_THREADPOOL

dnnl_status_t DNNL_API dnnl_stream_attr_set_threadpool(

        dnnl_stream_attr_t attr, void *threadpool);


dnnl_status_t DNNL_API dnnl_stream_attr_get_threadpool(

        dnnl_stream_attr_t attr, void **threadpool);

#endif


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);


#if DNNL_GPU_RUNTIME == DNNL_RUNTIME_OCL

dnnl_status_t DNNL_API dnnl_stream_create_ocl(

        dnnl_stream_t *stream, dnnl_engine_t engine, cl_command_queue queue);


dnnl_status_t DNNL_API dnnl_stream_get_ocl_command_queue(

        dnnl_stream_t stream, cl_command_queue *queue);

#endif


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_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_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);


#if DNNL_CPU_RUNTIME == DNNL_RUNTIME_THREADPOOL

dnnl_status_t DNNL_API dnnl_sgemm_tp(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,

        void *tp);


dnnl_status_t DNNL_API dnnl_gemm_u8s8s32_tp(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, void *tp);


dnnl_status_t DNNL_API dnnl_gemm_s8s8s32_tp(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, void *tp);

#endif


#ifdef __cplusplus

}

#endif


#endif

dnnl_types.h
C API types definitions.

dnnl_primitive_attr_set_rnn_data_qparams
dnnl_status_t DNNL_API dnnl_primitive_attr_set_rnn_data_qparams(dnnl_primitive_attr_t attr, const float scale, const float shift)
Set quantization scale and shift parameters for RNN data tensors.

dnnl_primitive_attr_t
struct dnnl_primitive_attr * dnnl_primitive_attr_t
A primitive descriptor attributes handle that controls primitive behavior.
Definition dnnl_types.h:1734

dnnl_post_ops_get_params_dw_k3s2p1
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)
Returns the parameters of an depthwise post-op with stride 2.

dnnl_primitive_attr_set_scratchpad_mode
dnnl_status_t DNNL_API dnnl_primitive_attr_set_scratchpad_mode(dnnl_primitive_attr_t attr, dnnl_scratchpad_mode_t mode)
Sets primitive attributes scratchpad mode.

dnnl_primitive_attr_get_post_ops
dnnl_status_t DNNL_API dnnl_primitive_attr_get_post_ops(const_dnnl_primitive_attr_t attr, const_dnnl_post_ops_t *post_ops)
Returns primitive attributes post-ops.

dnnl_post_ops_append_dw_k3s1p1
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)
Appends a depthwise post-op convolution with stride 1.

dnnl_post_ops_destroy
dnnl_status_t DNNL_API dnnl_post_ops_destroy(dnnl_post_ops_t post_ops)
Destroys post-ops.

dnnl_primitive_attr_set_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)
Sets primitive attributes zero points for primitive operations for a given memory argument.

dnnl_primitive_attr_set_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)
Sets primitive attributes post-ops.

dnnl_post_ops_append_sum
dnnl_status_t DNNL_API dnnl_post_ops_append_sum(dnnl_post_ops_t post_ops, float scale)
Appends an accumulation (sum) to post-ops.

dnnl_post_ops_t
struct dnnl_post_ops * dnnl_post_ops_t
A post operation chain handle.
Definition dnnl_types.h:1760

dnnl_primitive_attr_set_rnn_weights_qparams
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)
Sets quantization scaling factors for RNN weights tensors.

const_dnnl_primitive_attr_t
const struct dnnl_primitive_attr * const_dnnl_primitive_attr_t
A constant primitive descriptor attributes handle.
Definition dnnl_types.h:1737

dnnl_primitive_attr_destroy
dnnl_status_t DNNL_API dnnl_primitive_attr_destroy(dnnl_primitive_attr_t attr)
Destroys primitive attributes.

dnnl_post_ops_len
int DNNL_API dnnl_post_ops_len(const_dnnl_post_ops_t post_ops)
Returns the length of post-ops.

const_dnnl_post_ops_t
const struct dnnl_post_ops * const_dnnl_post_ops_t
A constant post operation chain handle.
Definition dnnl_types.h:1763

dnnl_post_ops_append_dw_k3s2p1
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)
Appends a depthwise post-op convolution with stride 2.

dnnl_post_ops_get_params_eltwise
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)
Returns the parameters of an elementwise post-up.

dnnl_post_ops_create
dnnl_status_t DNNL_API dnnl_post_ops_create(dnnl_post_ops_t *post_ops)
Creates empty post-ops sequence.

dnnl_primitive_attr_set_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)
Sets primitive attributes scaling factors for primitive operations for a given memory argument.

dnnl_primitive_attr_get_scratchpad_mode
dnnl_status_t DNNL_API dnnl_primitive_attr_get_scratchpad_mode(const_dnnl_primitive_attr_t attr, dnnl_scratchpad_mode_t *mode)
Returns the primitive attributes scratchpad mode.

dnnl_primitive_attr_clone
dnnl_status_t DNNL_API dnnl_primitive_attr_clone(dnnl_primitive_attr_t *attr, const_dnnl_primitive_attr_t existing_attr)
Clones primitive attributes.

dnnl_post_ops_get_params_dw_k3s1p1
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)
Returns the parameters of an depthwise post-op with stride 1.

dnnl_post_ops_get_kind
dnnl_primitive_kind_t DNNL_API dnnl_post_ops_get_kind(const_dnnl_post_ops_t post_ops, int index)
Returns the kind of a post-op entry.

dnnl_scratchpad_mode_t
dnnl_scratchpad_mode_t
Scratchpad mode.
Definition dnnl_types.h:1700

dnnl_post_ops_append_eltwise
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)
Appends an elementwise post-op.

dnnl_primitive_attr_get_zero_points
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)
Returns count, correspondence zero point mask, and a pointer to a constant int32_t array of zero_poin...

dnnl_post_ops_get_params_sum
dnnl_status_t DNNL_API dnnl_post_ops_get_params_sum(const_dnnl_post_ops_t post_ops, int index, float *scale)
Returns the parameters of an accumulation (sum) post-op.

dnnl_primitive_attr_set_output_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)
Sets output scaling factors correspondence mask and values.

dnnl_primitive_attr_get_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)
Returns primitive attributes scaling factors correspondence mask and values for a given memory argume...

dnnl_primitive_attr_create
dnnl_status_t DNNL_API dnnl_primitive_attr_create(dnnl_primitive_attr_t *attr)
Creates an empty (default) primitive attributes with all the parameters set to their default values.

dnnl_primitive_attr_get_output_scales
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)
Returns primitive attributes output scaling factors correspondence mask and values.

dnnl_batch_normalization_backward_desc_init
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)
Initializes a descriptor for a batch normalization backward propagation primitive.

dnnl_batch_normalization_forward_desc_init
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)
Initializes a descriptor for a batch normalization forward propagation primitive.

dnnl_binary_desc_init
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)
Initializes a descriptor for a binary primitive.

dnnl_gemm_s8s8s32
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)
Performs integer matrix-matrix multiply on 8-bit signed matrix A, 8-bit signed matrix B,...

dnnl_sgemm
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)
Performs single-precision matrix-matrix multiply.

dnnl_gemm_u8s8s32
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)
Performs integer matrix-matrix multiply on 8-bit unsigned matrix A, 8-bit signed matrix B,...

dnnl_concat_primitive_desc_create
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)
Creates a primitive descriptor for an out-of-place concatenation primitive.

dnnl_convolution_forward_desc_init
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)
Initializes a descriptor for a convolution forward propagation primitive.

dnnl_dilated_convolution_backward_weights_desc_init
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)
Initializes a descriptor for a dilated convolution weights gradient primitive.

dnnl_dilated_convolution_forward_desc_init
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)
Initializes a descriptor for a dilated convolution forward propagation primitive.

dnnl_convolution_backward_weights_desc_init
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)
Initializes a descriptor for a convolution weights gradient primitive.

dnnl_dilated_convolution_backward_data_desc_init
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)
Initializes a descriptor for a dilated convolution backward propagation primitive.

dnnl_convolution_backward_data_desc_init
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)
Initializes a descriptor for a convolution backward propagation primitive.

dnnl_dilated_deconvolution_backward_data_desc_init
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)
Initializes a descriptor for a dilated deconvolution backward propagation primitive.

dnnl_deconvolution_forward_desc_init
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)
Initializes a descriptor for a deconvolution forward propagation primitive.

dnnl_deconvolution_backward_weights_desc_init
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)
Initializes a descriptor for a deconvolution weights gradient primitive.

dnnl_deconvolution_backward_data_desc_init
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)
Initializes a descriptor for a deconvolution backward propagation primitive.

dnnl_dilated_deconvolution_forward_desc_init
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)
Initializes a descriptor for a dilated deconvolution forward propagation primitive.

dnnl_dilated_deconvolution_backward_weights_desc_init
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)
Initializes a descriptor for a dilated deconvolution weights gradient primitive.

dnnl_deconvolution_desc_t
dnnl_convolution_desc_t dnnl_deconvolution_desc_t
A descriptor of a deconvolution operation.
Definition dnnl_types.h:1179

dnnl_eltwise_forward_desc_init
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)
Initializes a descriptor for eltwise forward propagation primitive.

dnnl_eltwise_backward_desc_init
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)
Initializes a descriptor for eltwise backward propagation primitive.

dnnl_engine_kind_t
dnnl_engine_kind_t
Kinds of engines.
Definition dnnl_types.h:1646

dnnl_engine_get_ocl_device
dnnl_status_t DNNL_API dnnl_engine_get_ocl_device(dnnl_engine_t engine, cl_device_id *device)
Returns the OpenCL device associated with an engine.

dnnl_engine_t
struct dnnl_engine * dnnl_engine_t
An engine handle.
Definition dnnl_types.h:1659

dnnl_engine_create_ocl
dnnl_status_t DNNL_API dnnl_engine_create_ocl(dnnl_engine_t *engine, dnnl_engine_kind_t kind, cl_device_id device, cl_context context)
Creates an engine associated with an OpenCL device and an OpenCL context.

dnnl_engine_get_kind
dnnl_status_t DNNL_API dnnl_engine_get_kind(dnnl_engine_t engine, dnnl_engine_kind_t *kind)
Returns the kind of an engine.

dnnl_engine_destroy
dnnl_status_t DNNL_API dnnl_engine_destroy(dnnl_engine_t engine)
Destroys an engine.

dnnl_engine_get_ocl_context
dnnl_status_t DNNL_API dnnl_engine_get_ocl_context(dnnl_engine_t engine, cl_context *context)
Returns the OpenCL context associated with an engine.

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

dnnl_engine_get_count
size_t DNNL_API dnnl_engine_get_count(dnnl_engine_kind_t kind)
Returns the number of engines of a particular kind.

dnnl_inner_product_forward_desc_init
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)
Initializes descriptor for inner product forward propagation.

dnnl_inner_product_backward_weights_desc_init
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)
Initializes descriptor for inner product weights gradient primitive.

dnnl_inner_product_backward_data_desc_init
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)
Initializes descriptor for inner product backward propagation.

dnnl_layer_normalization_backward_desc_init
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)
Initializes a descriptor for a layer normalization backward propagation primitive.

dnnl_layer_normalization_forward_desc_init
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)
Initializes a descriptor for layer normalization forward propagation primitive.

dnnl_logsoftmax_desc_t
dnnl_softmax_desc_t dnnl_logsoftmax_desc_t
A descriptor of a LogSoftmax operation.
Definition dnnl_types.h:1283

dnnl_logsoftmax_forward_desc_init
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)
Initializes a descriptor for logsoftmax forward propagation primitive.

dnnl_logsoftmax_backward_desc_init
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)
Initializes a descriptor for logsoftmax backward propagation primitive.

dnnl_lrn_backward_desc_init
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)
Initializes a descriptor for LRN backward propagation primitive.

dnnl_lrn_forward_desc_init
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)
Initializes a descriptor for LRN forward propagation primitive.

dnnl_matmul_desc_init
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)
Initializes a matrix multiplication descriptor.

dnnl_data_type_t
dnnl_data_type_t
Data type specification.
Definition dnnl_types.h:62

const_dnnl_memory_t
const struct dnnl_memory * const_dnnl_memory_t
A constant memory handle.
Definition dnnl_types.h:1107

dnnl_memory_desc_init_submemory
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)
Initializes a memory descriptor for a region inside an area described by an existing memory descripto...

dnnl_memory_t
struct dnnl_memory * dnnl_memory_t
A memory handle.
Definition dnnl_types.h:1104

dnnl_format_tag_t
dnnl_format_tag_t
Memory format tag specification.
Definition dnnl_types.h:164

dnnl_memory_desc_permute_axes
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)
Initializes a memory descriptor by permuting axes in an existing one.

dnnl_memory_unmap_data
dnnl_status_t DNNL_API dnnl_memory_unmap_data(const_dnnl_memory_t memory, void *mapped_ptr)
Unmaps a memory object and writes back any changes made to the previously mapped memory buffer.

dnnl_memory_create
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)
Creates a memory object.

dnnl_memory_get_engine
dnnl_status_t DNNL_API dnnl_memory_get_engine(const_dnnl_memory_t memory, dnnl_engine_t *engine)
Returns the engine of a memory object.

dnnl_memory_desc_reshape
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)
Initializes a memory descriptor by reshaping an existing one.

dnnl_memory_get_memory_desc
dnnl_status_t DNNL_API dnnl_memory_get_memory_desc(const_dnnl_memory_t memory, const dnnl_memory_desc_t **memory_desc)
Returns the memory descriptor for a memory object.

dnnl_memory_set_data_handle
dnnl_status_t DNNL_API dnnl_memory_set_data_handle(dnnl_memory_t memory, void *handle)
Sets a memory object's data handle.

dnnl_memory_get_data_handle
dnnl_status_t DNNL_API dnnl_memory_get_data_handle(const_dnnl_memory_t memory, void **handle)
Returns memory object's data handle.

dnnl_memory_set_data_handle_v2
dnnl_status_t DNNL_API dnnl_memory_set_data_handle_v2(dnnl_memory_t memory, void *handle, dnnl_stream_t stream)
Sets a memory object's data handle.

dnnl_memory_desc_init_by_strides
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)
Initializes a memory descriptor using dimensions and strides.

dnnl_dims_t
dnnl_dim_t dnnl_dims_t[DNNL_MAX_NDIMS]
A type to describe tensor dimensions.
Definition dnnl_types.h:947

dnnl_dim_t
int64_t dnnl_dim_t
A type to describe tensor dimension.
Definition dnnl_types.h:944

dnnl_memory_destroy
dnnl_status_t DNNL_API dnnl_memory_destroy(dnnl_memory_t memory)
Destroys a memory object.

dnnl_memory_desc_equal
int DNNL_API dnnl_memory_desc_equal(const dnnl_memory_desc_t *lhs, const dnnl_memory_desc_t *rhs)
Compares two memory descriptors.

dnnl_memory_map_data
dnnl_status_t DNNL_API dnnl_memory_map_data(const_dnnl_memory_t memory, void **mapped_ptr)
Maps a memory object and returns a host-side pointer to a memory buffer with a copy of its contents.

dnnl_memory_set_ocl_mem_object
dnnl_status_t DNNL_API dnnl_memory_set_ocl_mem_object(dnnl_memory_t memory, cl_mem mem_object)
Sets OpenCL memory object associated with a memory object.

dnnl_memory_desc_get_size
size_t DNNL_API dnnl_memory_desc_get_size(const dnnl_memory_desc_t *memory_desc)
Returns the size of a memory descriptor.

dnnl_memory_get_ocl_mem_object
dnnl_status_t DNNL_API dnnl_memory_get_ocl_mem_object(const_dnnl_memory_t memory, cl_mem *mem_object)
Returns an OpenCL memory object associated with a memory object.

dnnl_memory_desc_init_by_tag
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)
Initializes a memory descriptor using dimensions and memory format tag.

dnnl_pooling_forward_desc_init
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)
Initializes a descriptor for pooling forward propagation primitive.

dnnl_pooling_backward_desc_init
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)
Initializes a descriptor for pooling backward propagation primitive.

dnnl_primitive_desc_query
dnnl_status_t DNNL_API dnnl_primitive_desc_query(const_dnnl_primitive_desc_t primitive_desc, dnnl_query_t what, int index, void *result)
Queries a primitive descriptor for various pieces of information.

dnnl_primitive_desc_iterator_destroy
dnnl_status_t DNNL_API dnnl_primitive_desc_iterator_destroy(dnnl_primitive_desc_iterator_t iterator)
Destroys a primitive descriptor iterator.

dnnl_primitive_t
struct dnnl_primitive * dnnl_primitive_t
A primitive handle.
Definition dnnl_types.h:1774

dnnl_primitive_desc_iterator_t
struct dnnl_primitive_desc_iterator * dnnl_primitive_desc_iterator_t
A primitive descriptor iterator handle.
Definition dnnl_types.h:1678

dnnl_primitive_desc_query_s32
int DNNL_API dnnl_primitive_desc_query_s32(const_dnnl_primitive_desc_t primitive_desc, dnnl_query_t what, int index)
Queries primitive descriptor for a signed 32bit int.

dnnl_primitive_desc_create
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)
Creates a primitive descriptor.

const_dnnl_primitive_t
const struct dnnl_primitive * const_dnnl_primitive_t
A constant primitive handle.
Definition dnnl_types.h:1776

dnnl_primitive_desc_get_attr
dnnl_status_t DNNL_API dnnl_primitive_desc_get_attr(const_dnnl_primitive_desc_t primitive_desc, const_dnnl_primitive_attr_t *attr)
Returns a constant reference to the attributes of a primitive descriptor.

dnnl_primitive_execute
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)
Executes a primitive.

dnnl_primitive_desc_iterator_next
dnnl_status_t DNNL_API dnnl_primitive_desc_iterator_next(dnnl_primitive_desc_iterator_t iterator)
Advances the primitive descriptor iterator to point to the next available implementation.

dnnl_primitive_desc_destroy
dnnl_status_t DNNL_API dnnl_primitive_desc_destroy(dnnl_primitive_desc_t primitive_desc)
Destroys a primitive descriptor.

const_dnnl_op_desc_t
const void * const_dnnl_op_desc_t
A pointer to any of the operation descriptors (constant variant).
Definition dnnl_types.h:1122

dnnl_primitive_get_primitive_desc
dnnl_status_t DNNL_API dnnl_primitive_get_primitive_desc(const_dnnl_primitive_t primitive, const_dnnl_primitive_desc_t *primitive_desc)
Retrieves a constant reference to the primitive descriptor of a given primitive.

dnnl_alg_kind_t
dnnl_alg_kind_t
Kinds of algorithms.
Definition dnnl_types.h:748

dnnl_primitive_kind_t
dnnl_primitive_kind_t
Kinds of primitives.
Definition dnnl_types.h:704

dnnl_query_t
dnnl_query_t
Primitive descriptor query specification.
Definition dnnl_types.h:2002

dnnl_primitive_desc_query_md
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)
Queries primitive descriptor for a memory descriptor.

dnnl_primitive_desc_t
struct dnnl_primitive_desc * dnnl_primitive_desc_t
A primitive descriptor handle.
Definition dnnl_types.h:1689

const_dnnl_primitive_desc_t
const struct dnnl_primitive_desc * const_dnnl_primitive_desc_t
A constant primitive descriptor handle.
Definition dnnl_types.h:1692

dnnl_primitive_destroy
dnnl_status_t DNNL_API dnnl_primitive_destroy(dnnl_primitive_t primitive)
Destroys a primitive.

dnnl_primitive_desc_iterator_create
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)
Creates a primitive descriptor iterator.

dnnl_primitive_create
dnnl_status_t DNNL_API dnnl_primitive_create(dnnl_primitive_t *primitive, const_dnnl_primitive_desc_t primitive_desc)
Creates a primitive.

dnnl_prop_kind_t
dnnl_prop_kind_t
Kinds of propagation.
Definition dnnl_types.h:677

dnnl_primitive_desc_clone
dnnl_status_t DNNL_API dnnl_primitive_desc_clone(dnnl_primitive_desc_t *primitive_desc, const_dnnl_primitive_desc_t existing_primitive_desc)
Clones a primitive descriptor.

dnnl_primitive_desc_iterator_fetch
dnnl_primitive_desc_t DNNL_API dnnl_primitive_desc_iterator_fetch(const_dnnl_primitive_desc_iterator_t iterator)
Fetches the current primitive descriptor from a primitive descriptor iterator.

const_dnnl_primitive_desc_iterator_t
const struct dnnl_primitive_desc_iterator * const_dnnl_primitive_desc_iterator_t
A constant primitive descriptor iterator handle.
Definition dnnl_types.h:1681

dnnl_reorder_primitive_desc_create
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)
Creates a primitive descriptor for a reorder primitive.

dnnl_resampling_backward_desc_init
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)
Initializes a descriptor for resampling backward propagation primitive.

dnnl_resampling_forward_desc_init
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)
Initializes a descriptor for a resampling forward propagation primitive.

dnnl_lbr_gru_backward_desc_init
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)
Initializes a descriptor for LBR GRU backward propagation primitive.

dnnl_gru_forward_desc_init
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)
Initializes a descriptor for GRU forward propagation primitive.

dnnl_vanilla_rnn_forward_desc_init
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)
Initializes a descriptor for vanilla RNN forward propagation primitive.

dnnl_lstm_backward_desc_init
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)
Initializes a descriptor for an LSTM backward propagation primitive.

dnnl_rnn_direction_t
dnnl_rnn_direction_t
A direction of RNN primitive execution.
Definition dnnl_types.h:1470

dnnl_vanilla_rnn_backward_desc_init
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)
Initializes a descriptor for vanilla RNN backward propagation primitive.

dnnl_lstm_backward_desc_init_v3
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)
Initializes a descriptor for an LSTM (with or without peephole and with or with out recurrent project...

dnnl_lstm_backward_desc_init_v2
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)
Initializes a descriptor for an LSTM (with or without peephole) backward propagation primitive.

dnnl_lstm_forward_desc_init
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)
Initializes a descriptor for LSTM forward propagation primitive.

dnnl_lbr_gru_forward_desc_init
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)
Initializes a descriptor for LBR GRU forward propagation primitive.

dnnl_lstm_forward_desc_init_v3
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)
Initializes a descriptor for an LSTM (with or without peephole and with or without recurrent projecti...

dnnl_lstm_forward_desc_init_v2
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)
Initializes a descriptor for an LSTM (with or without peephole) forward propagation primitive.

dnnl_gru_backward_desc_init
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)
Initializes a descriptor for GRU backward propagation primitive.

dnnl_set_jit_dump
dnnl_status_t DNNL_API dnnl_set_jit_dump(int enable)
Configures dumping of JIT-generated code.

dnnl_set_verbose
dnnl_status_t DNNL_API dnnl_set_verbose(int level)
Configures verbose output to stdout.

dnnl_cpu_isa_t
dnnl_cpu_isa_t
CPU instruction set flags.
Definition dnnl_types.h:2150

dnnl_set_max_cpu_isa
dnnl_status_t DNNL_API dnnl_set_max_cpu_isa(dnnl_cpu_isa_t isa)
Sets the maximal ISA the library can dispatch to on the CPU.

dnnl_set_jit_profiling_flags
dnnl_status_t DNNL_API dnnl_set_jit_profiling_flags(unsigned flags)
Sets library profiling flags.

dnnl_version
const dnnl_version_t DNNL_API * dnnl_version()
Returns library version information.

dnnl_set_jit_profiling_jitdumpdir
dnnl_status_t DNNL_API dnnl_set_jit_profiling_jitdumpdir(const char *dir)
Sets JIT dump output path.

dnnl_shuffle_forward_desc_init
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)
Initializes a descriptor for shuffle forward propagation primitive.

dnnl_shuffle_backward_desc_init
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)
Initializes a descriptor for shuffle backward propagation primitive.

dnnl_softmax_backward_desc_init
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)
Initializes a descriptor for softmax backward propagation primitive.

dnnl_softmax_forward_desc_init
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)
Initializes a descriptor for softmax forward propagation primitive.

dnnl_stream_attr_t
struct dnnl_stream_attr * dnnl_stream_attr_t
An execution stream attributes handle.
Definition dnnl_types.h:2091

dnnl_stream_create_ocl
dnnl_status_t DNNL_API dnnl_stream_create_ocl(dnnl_stream_t *stream, dnnl_engine_t engine, cl_command_queue queue)
Creates an execution stream for a given engine associated with an OpenCL command queue.

dnnl_stream_wait
dnnl_status_t DNNL_API dnnl_stream_wait(dnnl_stream_t stream)
Waits for all primitives in the execution stream to finish computations.

dnnl_stream_t
struct dnnl_stream * dnnl_stream_t
An execution stream handle.
Definition dnnl_types.h:2084

dnnl_stream_get_ocl_command_queue
dnnl_status_t DNNL_API dnnl_stream_get_ocl_command_queue(dnnl_stream_t stream, cl_command_queue *queue)
Returns the OpenCL command queue associated with an execution stream.

dnnl_stream_create_v2
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)
Creates an execution stream.

dnnl_stream_attr_destroy
dnnl_status_t DNNL_API dnnl_stream_attr_destroy(dnnl_stream_attr_t attr)
Destroys execution stream attributes.

dnnl_stream_destroy
dnnl_status_t DNNL_API dnnl_stream_destroy(dnnl_stream_t stream)
Destroys an execution stream.

const_dnnl_stream_attr_t
const struct dnnl_stream_attr * const_dnnl_stream_attr_t
A constant execution stream attributes handle.
Definition dnnl_types.h:2093

dnnl_stream_attr_create
dnnl_status_t DNNL_API dnnl_stream_attr_create(dnnl_stream_attr_t *attr, dnnl_engine_kind_t kind)
Creates execution stream attributes for a stream that runs on an engine of a particular kind.

dnnl_stream_create
dnnl_status_t DNNL_API dnnl_stream_create(dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags)
Creates an execution stream.

dnnl_sum_primitive_desc_create
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)
Creates a primitive descriptor for an (out-of-place) sum primitive.

dnnl_status_t
dnnl_status_t
Status values returned by the library functions.
Definition dnnl_types.h:39

dnnl_batch_normalization_desc_t
A descriptor of a Batch Normalization operation.
Definition dnnl_types.h:1360

dnnl_binary_desc_t
A descriptor of a binary operation.
Definition dnnl_types.h:1568

dnnl_convolution_desc_t
A descriptor of a convolution operation.
Definition dnnl_types.h:1134

dnnl_eltwise_desc_t
A descriptor of a element-wise operation.
Definition dnnl_types.h:1209

dnnl_exec_arg_t
A structure that contains an index and a memory object, and is used to pass arguments to dnnl_primiti...
Definition dnnl_types.h:1963

dnnl_inner_product_desc_t
A descriptor of an inner product operation.
Definition dnnl_types.h:1430

dnnl_layer_normalization_desc_t
A descriptor of a Layer Normalization operation.
Definition dnnl_types.h:1393

dnnl_lrn_desc_t
A descriptor of a Local Response Normalization (LRN) operation.
Definition dnnl_types.h:1329

dnnl_matmul_desc_t
A descriptor of a matrix multiplication operation.
Definition dnnl_types.h:1594

dnnl_memory_desc_t
Memory descriptor.
Definition dnnl_types.h:1050

dnnl_pooling_desc_t
A descriptor of a pooling operation.
Definition dnnl_types.h:1291

dnnl_resampling_desc_t
A descriptor of resampling operation.
Definition dnnl_types.h:1616

dnnl_rnn_desc_t
A descriptor for an RNN operation.
Definition dnnl_types.h:1486

dnnl_shuffle_desc_t
A descriptor of a shuffle operation.
Definition dnnl_types.h:1187

dnnl_softmax_desc_t
A descriptor of a Softmax operation.
Definition dnnl_types.h:1261

dnnl_version_t
Structure containing version information as per Semantic Versioning
Definition dnnl_types.h:2120