Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  0.16
Performance library for Deep Learning
mkldnn.h
Go to the documentation of this file.
1 /*******************************************************************************
2 * Copyright 2016-2018 Intel Corporation
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *******************************************************************************/
16 
17 #ifndef MKLDNN_H
18 #define MKLDNN_H
19 
20 #ifndef DOXYGEN_SHOULD_SKIP_THIS
21 
22 /* All symbols shall be internal unless marked as MKLDNN_API */
23 #if defined _WIN32 || defined __CYGWIN__
24 # define MKLDNN_HELPER_DLL_IMPORT __declspec(dllimport)
25 # define MKLDNN_HELPER_DLL_EXPORT __declspec(dllexport)
26 #else
27 # if __GNUC__ >= 4
28 # define MKLDNN_HELPER_DLL_IMPORT __attribute__ ((visibility ("default")))
29 # define MKLDNN_HELPER_DLL_EXPORT __attribute__ ((visibility ("default")))
30 # else
31 # define MKLDNN_HELPER_DLL_IMPORT
32 # define MKLDNN_HELPER_DLL_EXPORT
33 # endif
34 #endif
35 
36 #ifdef MKLDNN_DLL
37 # ifdef MKLDNN_DLL_EXPORTS
38 # define MKLDNN_API MKLDNN_HELPER_DLL_EXPORT
39 # else
40 # define MKLDNN_API MKLDNN_HELPER_DLL_IMPORT
41 # endif
42 #else
43 # define MKLDNN_API
44 #endif
45 
46 #if defined (__GNUC__)
47 # define MKLDNN_DEPRECATED __attribute__((deprecated))
48 #elif defined(_MSC_VER)
49 # define MKLDNN_DEPRECATED __declspec(deprecated)
50 #else
51 # define MKLDNN_DEPRECATED
52 #endif
53 
54 #include "mkldnn_types.h"
55 #endif /* DOXYGEN_SHOULD_SKIP_THIS */
56 
57 #ifdef __cplusplus
58 extern "C" {
59 #endif
60 
77  const_mkldnn_primitive_desc_t hint_forward_primitive_desc);
78 
87  mkldnn_engine_t engine,
88  const_mkldnn_primitive_desc_t hint_forward_primitive_desc);
89 
94 
102 
106 
112  mkldnn_primitive_desc_t *primitive_desc,
113  const_mkldnn_op_desc_t op_desc, mkldnn_engine_t engine,
114  const_mkldnn_primitive_desc_t hint_forward_primitive_desc);
115 
121  mkldnn_primitive_desc_t *primitive_desc,
123  mkldnn_engine_t engine,
124  const_mkldnn_primitive_desc_t hint_forward_primitive_desc);
125 
128  mkldnn_primitive_desc_t *primitive_desc,
129  const_mkldnn_primitive_desc_t existing_primitive_desc);
130 
140  const_mkldnn_primitive_desc_t primitive_desc,
142 
145  mkldnn_primitive_desc_t primitive_desc);
146 
151  const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what,
152  int index, void *result);
153 
158  const_mkldnn_primitive_desc_t primitive_desc);
159 
164  const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what,
165  int index);
166 
170 int MKLDNN_API mkldnn_primitive_desc_query_s32(
171  const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what,
172  int index);
173 
177  mkldnn_primitive_t *primitive,
178  const_mkldnn_primitive_desc_t primitive_desc,
179  const mkldnn_primitive_at_t *inputs,
180  const_mkldnn_primitive_t *outputs);
181 
189  const_mkldnn_primitive_t primitive,
190  const_mkldnn_primitive_desc_t *primitive_desc);
191 
194  const_mkldnn_primitive_t primitive, size_t index,
195  mkldnn_primitive_at_t *input);
196 
199  const_mkldnn_primitive_t primitive, size_t index,
200  const_mkldnn_primitive_t *output);
201 
204  mkldnn_primitive_t primitive);
205 
212  const_mkldnn_primitive_t primitive, size_t output_index);
213 
228 
232  const_mkldnn_primitive_attr_t existing_attr);
233 
237 
242 
250 
264  const_mkldnn_primitive_attr_t attr, int *count, int *mask,
265  const float **scales);
266 
306  mkldnn_primitive_attr_t attr, int count, int mask,
307  const float *scales);
308 
319 
331 
338 
341 
343 int MKLDNN_API mkldnn_post_ops_len(const_mkldnn_post_ops_t post_ops);
344 
348  const_mkldnn_post_ops_t post_ops, int index);
349 
371  mkldnn_post_ops_t post_ops, float scale);
372 
381  const_mkldnn_post_ops_t post_ops, int index, float *scale);
382 
395  mkldnn_post_ops_t post_ops, float scale, mkldnn_alg_kind_t alg,
396  float alpha, float beta);
397 
402  const_mkldnn_post_ops_t post_ops, int index, float *scale,
403  mkldnn_alg_kind_t *alg, float *alpha, float *beta);
404 
417  mkldnn_memory_desc_t *memory_desc, int ndims, const mkldnn_dims_t dims,
418  mkldnn_data_type_t data_type, mkldnn_memory_format_t format);
419 
424  mkldnn_primitive_desc_t *memory_primitive_desc,
425  const mkldnn_memory_desc_t *memory_desc, mkldnn_engine_t engine);
426 
431  mkldnn_primitive_desc_t *view_primitive_desc,
432  const_mkldnn_primitive_desc_t memory_primitive_desc,
433  const mkldnn_dims_t dims, const mkldnn_dims_t offsets);
434 
445 
448 /* XXX: view? */
449 size_t MKLDNN_API mkldnn_memory_primitive_desc_get_size(
450  const_mkldnn_primitive_desc_t memory_primitive_desc);
451 
454 /* XXX: view? */
456  const_mkldnn_primitive_t memory, void **handle);
457 
460  mkldnn_primitive_t memory, void *handle);
461 
471  mkldnn_primitive_desc_t *reorder_primitive_desc,
474 
478  mkldnn_primitive_desc_t *reorder_primitive_desc,
482 
495  mkldnn_primitive_desc_t *concat_primitive_desc,
496  const mkldnn_memory_desc_t *output_desc, int n, int concat_dimension,
497  const_mkldnn_primitive_desc_t *input_pds);
498 
499 #if 0
500 
510 mkldnn_status_t MKLDNN_API mkldnn_concat_inplace_by_input_primitive_desc_create(
511  mkldnn_primitive_desc_t *concat_primitive_desc,
512  int n, int concat_dimension, const_mkldnn_primitive_desc_t *inputs);
513 
522 mkldnn_status_t MKLDNN_API mkldnn_concat_inplace_by_output_primitive_desc_create(
523  mkldnn_primitive_desc_t *concat_primitive_desc,
524  const mkldnn_primitive_desc_t output, int n, int concat_dimension,
525  int *sizes);
526 #endif
527 
540  mkldnn_primitive_desc_t *sum_primitive_desc,
541  const mkldnn_memory_desc_t *output_desc, int n, const float *scales,
542  const_mkldnn_primitive_desc_t *input_pds);
543 
579  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc,
580  const mkldnn_memory_desc_t *weights_desc,
581  const mkldnn_memory_desc_t *bias_desc,
582  const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides,
583  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
585 
600  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc,
601  const mkldnn_memory_desc_t *weights_desc,
602  const mkldnn_memory_desc_t *bias_desc,
603  const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides,
604  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
606 
614  mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
615  const mkldnn_memory_desc_t *diff_src_desc,
616  const mkldnn_memory_desc_t *weights_desc,
617  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
618  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
620 
628  mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
629  const mkldnn_memory_desc_t *diff_src_desc,
630  const mkldnn_memory_desc_t *weights_desc,
631  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
632  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
634 
642  mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
643  const mkldnn_memory_desc_t *src_desc,
644  const mkldnn_memory_desc_t *diff_weights_desc,
645  const mkldnn_memory_desc_t *diff_bias_desc,
646  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
647  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
649 
656 mkldnn_status_t MKLDNN_API
658  mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
659  const mkldnn_memory_desc_t *src_desc,
660  const mkldnn_memory_desc_t *diff_weights_desc,
661  const mkldnn_memory_desc_t *diff_bias_desc,
662  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
663  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
665 
687  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc,
688  const mkldnn_memory_desc_t *weights_desc,
689  const mkldnn_memory_desc_t *bias_desc,
690  const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides,
691  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
693 
708  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc,
709  const mkldnn_memory_desc_t *weights_desc,
710  const mkldnn_memory_desc_t *bias_desc,
711  const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides,
712  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
714 
722  mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
723  const mkldnn_memory_desc_t *diff_src_desc,
724  const mkldnn_memory_desc_t *weights_desc,
725  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
726  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
728 
736  mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
737  const mkldnn_memory_desc_t *diff_src_desc,
738  const mkldnn_memory_desc_t *weights_desc,
739  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
740  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
742 
750  mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
751  const mkldnn_memory_desc_t *src_desc,
752  const mkldnn_memory_desc_t *diff_weights_desc,
753  const mkldnn_memory_desc_t *diff_bias_desc,
754  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
755  const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r,
757 
765  mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind,
766  const mkldnn_memory_desc_t *src_desc,
767  const mkldnn_memory_desc_t *diff_weights_desc,
768  const mkldnn_memory_desc_t *diff_bias_desc,
769  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
770  const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l,
772 
798  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc,
799  float alpha, float beta);
800 
806  mkldnn_eltwise_desc_t *eltwise_desc, mkldnn_alg_kind_t alg_kind,
807  const mkldnn_memory_desc_t *diff_data_desc,
808  const mkldnn_memory_desc_t *data_desc, float alpha, float beta);
809 
825 MKLDNN_DEPRECATED
828  const mkldnn_memory_desc_t *data_desc, float negative_slope);
829 
835 MKLDNN_DEPRECATED
837  mkldnn_relu_desc_t *relu_desc,
838  const mkldnn_memory_desc_t *diff_data_desc,
839  const mkldnn_memory_desc_t *data_desc, float negative_slope);
840 
860  const mkldnn_memory_desc_t *data_desc, int softmax_axis);
861 
865  mkldnn_softmax_desc_t *softmax_desc,
866  const mkldnn_memory_desc_t *diff_desc,
867  const mkldnn_memory_desc_t *data_desc, int softmax_axis);
868 
900  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc,
901  const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides,
902  const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l,
904 
912  mkldnn_pooling_desc_t *pool_desc, mkldnn_alg_kind_t alg_kind,
913  const mkldnn_memory_desc_t *diff_src_desc,
914  const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides,
915  const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l,
917 
945  mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc,
946  int local_size, float alpha, float beta, float k);
947 
952  mkldnn_lrn_desc_t *lrn_desc, mkldnn_alg_kind_t alg_kind,
953  const mkldnn_memory_desc_t *diff_data_desc,
954  const mkldnn_memory_desc_t *data_desc, int local_size, float alpha,
955  float beta, float k);
956 
989  float epsilon, unsigned flags);
990 
1001  const mkldnn_memory_desc_t *diff_data_desc,
1002  const mkldnn_memory_desc_t *data_desc,
1003  float epsilon, unsigned flags);
1004 
1029  const mkldnn_memory_desc_t *src_desc,
1030  const mkldnn_memory_desc_t *weights_desc,
1031  const mkldnn_memory_desc_t *bias_desc,
1032  const mkldnn_memory_desc_t *dst_desc);
1033 
1041  mkldnn_inner_product_desc_t *ip_desc,
1042  const mkldnn_memory_desc_t *diff_src_desc,
1043  const mkldnn_memory_desc_t *weights_desc,
1044  const mkldnn_memory_desc_t *diff_dst_desc);
1045 
1053  mkldnn_inner_product_desc_t *ip_desc,
1054  const mkldnn_memory_desc_t *src_desc,
1055  const mkldnn_memory_desc_t *diff_weights_desc,
1056  const mkldnn_memory_desc_t *diff_bias_desc,
1057  const mkldnn_memory_desc_t *diff_dst_desc);
1058 
1072  mkldnn_convolution_relu_desc_t *conv_relu_desc,
1073  const mkldnn_convolution_desc_t *conv_desc, float negative_slope);
1074 
1091  mkldnn_rnn_cell_desc_t *rnn_cell_desc,
1093  unsigned int flags, float alpha, float clipping);
1094 
1096 int MKLDNN_API mkldnn_rnn_cell_get_gates_count(
1097  const mkldnn_rnn_cell_desc_t *rnn_cell_desc);
1098 
1100 int MKLDNN_API mkldnn_rnn_cell_get_states_count(
1101  const mkldnn_rnn_cell_desc_t *rnn_cell_desc);
1102 
1112  const mkldnn_rnn_cell_desc_t *rnn_cell_desc,
1113  const mkldnn_rnn_direction_t direction,
1114  const mkldnn_memory_desc_t *src_layer_desc,
1115  const mkldnn_memory_desc_t *src_iter_desc,
1116  const mkldnn_memory_desc_t *weights_layer_desc,
1117  const mkldnn_memory_desc_t *weights_iter_desc,
1118  const mkldnn_memory_desc_t *bias_desc,
1119  const mkldnn_memory_desc_t *dst_layer_desc,
1120  const mkldnn_memory_desc_t *dst_iter_desc);
1121 
1128  const mkldnn_rnn_cell_desc_t *rnn_cell_desc,
1129  const mkldnn_rnn_direction_t direction,
1130  const mkldnn_memory_desc_t *src_layer_desc,
1131  const mkldnn_memory_desc_t *src_iter_desc,
1132  const mkldnn_memory_desc_t *weights_layer_desc,
1133  const mkldnn_memory_desc_t *weights_iter_desc,
1134  const mkldnn_memory_desc_t *bias_desc,
1135  const mkldnn_memory_desc_t *dst_layer_desc,
1136  const mkldnn_memory_desc_t *dst_iter_desc,
1137  const mkldnn_memory_desc_t *diff_src_layer_desc,
1138  const mkldnn_memory_desc_t *diff_src_iter_desc,
1139  const mkldnn_memory_desc_t *diff_weights_layer_desc,
1140  const mkldnn_memory_desc_t *diff_weights_iter_desc,
1141  const mkldnn_memory_desc_t *diff_bias_desc,
1142  const mkldnn_memory_desc_t *diff_dst_layer,
1143  const mkldnn_memory_desc_t *diff_dst_iter_desc);
1144 
1153 size_t MKLDNN_API mkldnn_engine_get_count(mkldnn_engine_kind_t kind);
1154 
1157  mkldnn_engine_kind_t kind, size_t index);
1158 
1161  mkldnn_engine_kind_t *kind);
1162 
1165 
1173  mkldnn_stream_kind_t stream_kind);
1174 
1179  size_t n, mkldnn_primitive_t primitives[],
1180  mkldnn_primitive_t *error_primitive);
1181 
1186  int block, mkldnn_primitive_t *error_primitive);
1187 
1191  mkldnn_primitive_t *error_primitive);
1192 
1195 
1209 mkldnn_status_t MKLDNN_API mkldnn_verbose_set(int level);
1210 
1227 mkldnn_status_t MKLDNN_API mkldnn_sgemm(const char *transa, const char *transb,
1228  const int *M, const int *N, const int *K,
1229  const float *alpha, const float *A, const int *lda,
1230  const float *B, const int *ldb,
1231  const float *beta, float *C, const int *ldc);
1232 
1237 #ifdef __cplusplus
1238 }
1239 #endif
1240 
1241 #endif
A descriptor of a Local Response Normalization (LRN) operation.
Definition: mkldnn_types.h:746
mkldnn_status_t MKLDNN_API mkldnn_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *diff_weights_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a convolution descriptor conv_desc for backward propagation with respect to weights using...
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_destroy(mkldnn_primitive_attr_t attr)
Deletes an attr.
mkldnn_status_t MKLDNN_API mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc_t *sum_primitive_desc, const mkldnn_memory_desc_t *output_desc, int n, const float *scales, const_mkldnn_primitive_desc_t *input_pds)
Creates out-of-place sum_primitive_desc for sum of n inputs multiplied by scale with resulting output...
mkldnn_status_t MKLDNN_API mkldnn_stream_destroy(mkldnn_stream_t stream)
Destroys an execution stream.
mkldnn_status_t
Status values returned by Intel(R) MKL-DNN functions.
Definition: mkldnn_types.h:39
A descriptor of a convolution operation.
Definition: mkldnn_types.h:609
mkldnn_status_t MKLDNN_API mkldnn_memory_primitive_desc_create(mkldnn_primitive_desc_t *memory_primitive_desc, const mkldnn_memory_desc_t *memory_desc, mkldnn_engine_t engine)
Creates a memory_primitive_desc memory primitive descriptor using memory_desc and engine...
mkldnn_status_t MKLDNN_API mkldnn_post_ops_create(mkldnn_post_ops_t *post_ops)
Creates an empty sequence of post operations post_ops.
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_destroy(mkldnn_primitive_desc_t primitive_desc)
Deletes a primitive_desc.
mkldnn_status_t MKLDNN_API mkldnn_concat_primitive_desc_create(mkldnn_primitive_desc_t *concat_primitive_desc, const mkldnn_memory_desc_t *output_desc, int n, int concat_dimension, const_mkldnn_primitive_desc_t *input_pds)
Creates out-of-place concat_primitive_desc for concatenation of n inputs by concat_dimension with res...
padding_kind
Definition: mkldnn.hpp:229
An opaque structure to describe an engine.
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_get_attr(const_mkldnn_primitive_desc_t primitive_desc, const_mkldnn_primitive_attr_t *attr)
Returns a constant reference to the attribute of a primitive_desc.
mkldnn_status_t MKLDNN_API mkldnn_memory_desc_init(mkldnn_memory_desc_t *memory_desc, int ndims, const mkldnn_dims_t dims, mkldnn_data_type_t data_type, mkldnn_memory_format_t format)
Initializes a memory_desc memory descriptor using ndims, dims, data_type, and data format...
mkldnn_status_t MKLDNN_API mkldnn_dilated_convolution_backward_weights_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *diff_weights_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a convolution descriptor conv_desc for backward propagation with respect to weights using...
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.
A descriptor of a Softmax operation.
Definition: mkldnn_types.h:696
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.
mkldnn_status_t MKLDNN_API mkldnn_memory_get_data_handle(const_mkldnn_primitive_t memory, void **handle)
For a memory primitive, returns the data handle.
mkldnn_status_t MKLDNN_API mkldnn_convolution_backward_data_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a convolution descriptor conv_desc for backward propagation with respect to data using al...
A descriptor of an inner product operation.
Definition: mkldnn_types.h:804
mkldnn_status_t MKLDNN_API mkldnn_post_ops_destroy(mkldnn_post_ops_t post_ops)
Deletes a post_ops sequence.
An opaque structure for a chain of post operations.
An opaque structure to describe a primitive descriptor .
mkldnn_rnn_direction_t
A direction of RNN primitive execution.
Definition: mkldnn_types.h:869
mkldnn_status_t MKLDNN_API mkldnn_memory_set_data_handle(mkldnn_primitive_t memory, void *handle)
For a memory primitive, sets the data handle.
mkldnn_status_t MKLDNN_API mkldnn_batch_normalization_forward_desc_init(mkldnn_batch_normalization_desc_t *bnrm_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, float epsilon, unsigned flags)
Initializes a batch normalization descriptor bnrm_desc for forward propagation using prop_kind...
mkldnn_status_t MKLDNN_API mkldnn_post_ops_append_eltwise(mkldnn_post_ops_t post_ops, float scale, mkldnn_alg_kind_t alg, float alpha, float beta)
Appends eltwise post operation to the post_ops with given parameters kind, alpha and beta (...
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_backward_data_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a deconvolution descriptor conv_desc for backward propagation with respect to data using ...
mkldnn_primitive_at_t MKLDNN_API mkldnn_primitive_at(const_mkldnn_primitive_t primitive, size_t output_index)
Creates an mkldnn_primitive_at_t structure from a primitive and output_index.
mkldnn_status_t MKLDNN_API mkldnn_softmax_forward_desc_init(mkldnn_softmax_desc_t *softmax_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, int softmax_axis)
Initializes a softmax_desc for forward propagation using prop_kind (possible value are mkldnn_forward...
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_iterator_next(mkldnn_primitive_desc_iterator_t iterator)
Iterates over primitive descriptors.
mkldnn_status_t MKLDNN_API mkldnn_pooling_backward_desc_init(mkldnn_pooling_desc_t *pool_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a pooling descriptor pool_desc for backward propagation using alg_kind, memory descriptors, and pooling parameters in spatial domain: strides, kernel sizes, padding_l, padding_r, and padding_kind.
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_create(mkldnn_primitive_attr_t *attr)
Creates an empty (default) attr attribute.
A descriptor of a convolution followed by relu operation.
Definition: mkldnn_types.h:833
mkldnn_status_t MKLDNN_API mkldnn_stream_submit(mkldnn_stream_t stream, size_t n, mkldnn_primitive_t primitives[], mkldnn_primitive_t *error_primitive)
Submits primitives to an execution stream.
mkldnn_status_t MKLDNN_API mkldnn_rnn_cell_desc_init(mkldnn_rnn_cell_desc_t *rnn_cell_desc, mkldnn_alg_kind_t kind, mkldnn_alg_kind_t f, unsigned int flags, float alpha, float clipping)
Initializes a recurrent cell descriptor rnn_cell_desc using rnn_cell_desc, kind (possible values are ...
A descriptor of a element-wise operation.
Definition: mkldnn_types.h:654
mkldnn_status_t MKLDNN_API mkldnn_stream_wait(mkldnn_stream_t stream, int block, mkldnn_primitive_t *error_primitive)
Waits for all primitives in the execution stream to finish.
A descriptor for an rnn operation.
Definition: mkldnn_types.h:884
size_t MKLDNN_API mkldnn_engine_get_count(mkldnn_engine_kind_t kind)
Returns the number of engines of a particular kind.
mkldnn_primitive_kind_t MKLDNN_API mkldnn_post_ops_get_kind(const_mkldnn_post_ops_t post_ops, int index)
Returns the type of post operation with index index in given post_ops.
Memory descriptor.
Definition: mkldnn_types.h:583
mkldnn_status_t MKLDNN_API mkldnn_inner_product_backward_data_desc_init(mkldnn_inner_product_desc_t *ip_desc, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *diff_dst_desc)
Initializes an inner product descriptor ip_desc for backward propagation with respect to data using m...
const mkldnn_memory_desc_t MKLDNN_API * mkldnn_primitive_desc_query_memory_d(const_mkldnn_primitive_desc_t primitive_desc)
Queries primitive descriptor for memory descriptor.
prop_kind
Definition: mkldnn.hpp:237
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_forward_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a dilated deconvolution descriptor deconv_desc for forward propagation using prop_kind (p...
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.
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_get_int_output_round_mode(const_mkldnn_primitive_attr_t attr, mkldnn_round_mode_t *round_mode)
Returns integer output rounding mode round_mode for a given attr, previously set by mkldnn_primitive_...
mkldnn_round_mode_t
Rounding mode.
Definition: mkldnn_types.h:78
mkldnn_status_t MKLDNN_API mkldnn_convolution_forward_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a convolution descriptor conv_desc for forward propagation using prop_kind (possible valu...
mkldnn_status_t MKLDNN_API mkldnn_view_primitive_desc_create(mkldnn_primitive_desc_t *view_primitive_desc, const_mkldnn_primitive_desc_t memory_primitive_desc, const mkldnn_dims_t dims, const mkldnn_dims_t offsets)
Creates a view_primitive_desc for a given memory_primitive_desc, with dims sizes and offset offsets...
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_get_output_scales(const_mkldnn_primitive_attr_t attr, int *count, int *mask, const float **scales)
Returns count, correspondence scale mask, and pointer to a constant floating point array of output sc...
round_mode
Definition: mkldnn.hpp:220
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_set_post_ops(mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t post_ops)
Sets configured post_ops to an attribute attr for future use (when primitive descriptor is being crea...
mkldnn_memory_format_t
Memory format specification.
Definition: mkldnn_types.h:107
mkldnn_status_t MKLDNN_API mkldnn_eltwise_forward_desc_init(mkldnn_eltwise_desc_t *eltwise_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc, float alpha, float beta)
Initializes a eltwise_desc for forward propagation using prop_kind (possible values are mkldnn_forwar...
int MKLDNN_API mkldnn_memory_primitive_desc_equal(const_mkldnn_primitive_desc_t lhs, const_mkldnn_primitive_desc_t rhs)
Compares two descriptors of memory primitives.
mkldnn_status_t MKLDNN_API mkldnn_verbose_set(int level)
Sets verbosity level (print information to stdout).
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_forward_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a deconvolution descriptor deconv_desc for forward propagation using prop_kind (possible ...
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_query(const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index, void *result)
Queries primitive descriptor.
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_backward_weights_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *diff_weights_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a dilated deconvolution descriptor conv_desc for backward propagation with respect to wei...
An opaque structure to describe an execution stream.
mkldnn_status_t MKLDNN_API mkldnn_eltwise_backward_desc_init(mkldnn_eltwise_desc_t *eltwise_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, float alpha, float beta)
Initializes a eltwise_desc for backward propagation using alg_kind algorithm memory descriptors diff_...
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_iterator_destroy(mkldnn_primitive_desc_iterator_t iterator)
Deletes a primitive descriptor iterator.
mkldnn_status_t MKLDNN_API mkldnn_post_ops_append_sum(mkldnn_post_ops_t post_ops, float scale)
Appends accumulation (sum) post operation to the post_ops.
mkldnn_status_t MKLDNN_API mkldnn_primitive_get_output(const_mkldnn_primitive_t primitive, size_t index, const_mkldnn_primitive_t *output)
For a primitive, returns output at the index position.
mkldnn_prop_kind_t
Kinds of propagation.
Definition: mkldnn_types.h:341
mkldnn_stream_kind_t
Kinds of streams.
Definition: mkldnn_types.h:1131
mkldnn_status_t MKLDNN_API mkldnn_primitive_get_input_at(const_mkldnn_primitive_t primitive, size_t index, mkldnn_primitive_at_t *input)
For a primitive, returns input at the index position.
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_set_int_output_round_mode(mkldnn_primitive_attr_t attr, mkldnn_round_mode_t round_mode)
Sets output rounding mode round_mode for integer operations for a given attr.
A wrapper structure to specify a particular output of a primitive.
Definition: mkldnn_types.h:1043
mkldnn_status_t MKLDNN_API mkldnn_softmax_backward_desc_init(mkldnn_softmax_desc_t *softmax_desc, const mkldnn_memory_desc_t *diff_desc, const mkldnn_memory_desc_t *data_desc, int softmax_axis)
Initializes a softmax_desc for backward propagation using memory descriptors diff_desc and data_desc...
An opaque structure for primitive descriptor attributes.
mkldnn_status_t MKLDNN_API mkldnn_pooling_forward_desc_init(mkldnn_pooling_desc_t *pool_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t kernel, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a pooling descriptor pool_desc for forward propagation using prop_kind (possible values a...
mkldnn_status_t MKLDNN_API mkldnn_stream_create(mkldnn_stream_t *stream, mkldnn_stream_kind_t stream_kind)
Creates an execution stream of stream_kind.
MKLDNN_DEPRECATED mkldnn_status_t MKLDNN_API mkldnn_relu_backward_desc_init(mkldnn_relu_desc_t *relu_desc, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, float negative_slope)
Initializes a relu_desc for backward propagation using negative_slope and memory descriptors diff_dat...
MKLDNN_DEPRECATED mkldnn_status_t MKLDNN_API mkldnn_relu_forward_desc_init(mkldnn_relu_desc_t *relu_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *data_desc, float negative_slope)
Initializes a relu_desc for forward propagation using prop_kind (possible values are mkldnn_forward_t...
mkldnn_status_t MKLDNN_API mkldnn_batch_normalization_backward_desc_init(mkldnn_batch_normalization_desc_t *bnrm_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, float epsilon, unsigned flags)
Initializes a batch normalization descriptor bnrm_desc for backward propagation with respect to data ...
mkldnn_status_t MKLDNN_API mkldnn_primitive_get_primitive_desc(const_mkldnn_primitive_t primitive, const_mkldnn_primitive_desc_t *primitive_desc)
Retrieves a reference to the primitive_desc descriptor of given primitive.
mkldnn_status_t MKLDNN_API mkldnn_post_ops_get_params_eltwise(const_mkldnn_post_ops_t post_ops, int index, float *scale, mkldnn_alg_kind_t *alg, float *alpha, float *beta)
Gets the eltwise parameters of the post operation with index index in the sequence of post_ops...
mkldnn_status_t MKLDNN_API mkldnn_post_ops_get_params_sum(const_mkldnn_post_ops_t post_ops, int index, float *scale)
Gets the parameters of the accumulation (sum) post operation with index index in the sequence of post...
mkldnn_query_t
Primitive descriptor query specification.
Definition: mkldnn_types.h:1081
A descriptor of a Batch Normalization operation.
Definition: mkldnn_types.h:773
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_create(mkldnn_primitive_desc_t *primitive_desc, const_mkldnn_op_desc_t op_desc, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
Creates a primitive_desc using op_desc, engine, and optionally a hint primitive descriptor from forwa...
mkldnn_status_t MKLDNN_API mkldnn_reorder_primitive_desc_create_v2(mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output, const_mkldnn_primitive_attr_t attr)
Initializes a reorder_primitive_desc using an attr attribute and descriptors of input and output memo...
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_iterator_create(mkldnn_primitive_desc_iterator_t *iterator, const_mkldnn_op_desc_t op_desc, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
Creates a primitive descriptor iterator for given op_desc, engine, and optionally a hint primitive de...
mkldnn_status_t MKLDNN_API mkldnn_dilated_deconvolution_backward_data_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a dilated deconvolution descriptor conv_desc for backward propagation with respect to dat...
mkldnn_status_t MKLDNN_API mkldnn_stream_rerun(mkldnn_stream_t stream, mkldnn_primitive_t *error_primitive)
Reruns all the primitives within the stream.
A descriptor of a pooling operation.
Definition: mkldnn_types.h:712
mkldnn_primitive_desc_t MKLDNN_API mkldnn_primitive_desc_iterator_fetch(const_mkldnn_primitive_desc_iterator_t iterator)
Fetches current primitive descriptor.
int MKLDNN_API mkldnn_primitive_desc_query_s32(const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index)
Queries primitive descriptor for signed 32bit int.
mkldnn_status_t MKLDNN_API mkldnn_reorder_primitive_desc_create(mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output)
Initializes a reorder_primitive_desc using descriptors of input and output memory primitives...
int MKLDNN_API mkldnn_rnn_cell_get_states_count(const mkldnn_rnn_cell_desc_t *rnn_cell_desc)
Returns the number of states of a particular rnn_cell_desc.
mkldnn_primitive_kind_t
Kinds of primitives.
Definition: mkldnn_types.h:368
size_t MKLDNN_API mkldnn_memory_primitive_desc_get_size(const_mkldnn_primitive_desc_t memory_primitive_desc)
Returns the size (in bytes) that is required for given memory_primitive_desc.
mkldnn_status_t MKLDNN_API mkldnn_sgemm(const char *transa, const char *transb, const int *M, const int *N, const int *K, const float *alpha, const float *A, const int *lda, const float *B, const int *ldb, const float *beta, float *C, const int *ldc)
SGEMM performs matrix-matrix multiplication operation C := alpha*op( A )*op( B ) + beta*C...
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_get_post_ops(const_mkldnn_primitive_attr_t attr, const_mkldnn_post_ops_t *post_ops)
Returns post_ops for given attr.
mkldnn_status_t MKLDNN_API mkldnn_primitive_create(mkldnn_primitive_t *primitive, const_mkldnn_primitive_desc_t primitive_desc, const mkldnn_primitive_at_t *inputs, const_mkldnn_primitive_t *outputs)
Creates a primitive using a primitive_desc descriptor and arrays of inputs and outputs.
Definition: mkldnn_types.h:850
mkldnn_status_t MKLDNN_API mkldnn_dilated_convolution_forward_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a dilated convolution descriptor conv_desc for forward propagation using prop_kind (possi...
mkldnn_status_t MKLDNN_API mkldnn_engine_get_kind(mkldnn_engine_t engine, mkldnn_engine_kind_t *kind)
Returns the kind of an engine.
mkldnn_status_t MKLDNN_API mkldnn_engine_destroy(mkldnn_engine_t engine)
Destroys an engine.
mkldnn_status_t MKLDNN_API mkldnn_dilated_convolution_backward_data_desc_init(mkldnn_convolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t dilates, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a dilated convolution descriptor conv_desc for backward propagation with respect to data ...
mkldnn_status_t MKLDNN_API mkldnn_lrn_backward_desc_init(mkldnn_lrn_desc_t *lrn_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, int local_size, float alpha, float beta, float k)
Initializes an lrn_desc for backward propagation using alg_kind, memory descriptors data_desc...
mkldnn_padding_kind_t
Kinds of padding.
Definition: mkldnn_types.h:335
const_mkldnn_primitive_desc_t MKLDNN_API mkldnn_primitive_desc_query_pd(const_mkldnn_primitive_desc_t primitive_desc, mkldnn_query_t what, int index)
Queries primitive descriptor for primitive descriptor.
An opaque structure to describe a primitive.
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_create_v2(mkldnn_primitive_desc_t *primitive_desc, const_mkldnn_op_desc_t op_desc, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
Creates a primitive_desc using op_desc, attr, engine, and optionally a hint primitive descriptor from...
mkldnn_data_type_t
Data type specification.
Definition: mkldnn_types.h:62
mkldnn_status_t MKLDNN_API mkldnn_lrn_forward_desc_init(mkldnn_lrn_desc_t *lrn_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc, int local_size, float alpha, float beta, float k)
Initializes an lrn_desc for forward propagation using prop_kind (possible values are mkldnn_forward_t...
mkldnn_engine_kind_t
Kinds of engines.
Definition: mkldnn_types.h:931
int MKLDNN_API mkldnn_rnn_cell_get_gates_count(const mkldnn_rnn_cell_desc_t *rnn_cell_desc)
Returns the number of gates of a particular rnn_cell_desc.
const void * const_mkldnn_op_desc_t
A pointer to any of the operation descriptors (constant variant).
Definition: mkldnn_types.h:577
mkldnn_status_t MKLDNN_API mkldnn_primitive_desc_iterator_create_v2(mkldnn_primitive_desc_iterator_t *iterator, const_mkldnn_op_desc_t op_desc, const_mkldnn_primitive_attr_t attr, mkldnn_engine_t engine, const_mkldnn_primitive_desc_t hint_forward_primitive_desc)
Creates a primitive descriptor iterator for given op_desc, attr, engine, and optionally a hint primit...
mkldnn_status_t MKLDNN_API mkldnn_convolution_relu_desc_init(mkldnn_convolution_relu_desc_t *conv_relu_desc, const mkldnn_convolution_desc_t *conv_desc, float negative_slope)
Initializes a merged convolution-relu descriptor conv_relu_desc for forward propagation (supported in...
int MKLDNN_API mkldnn_post_ops_len(const_mkldnn_post_ops_t post_ops)
Returns the length of post operations for given post_ops.
mkldnn_alg_kind_t
Kinds of algorithms.
Definition: mkldnn_types.h:408
int mkldnn_dims_t[TENSOR_MAX_DIMS]
A type to describe tensor dimensions.
Definition: mkldnn_types.h:525
mkldnn_status_t MKLDNN_API mkldnn_primitive_destroy(mkldnn_primitive_t primitive)
Deletes a primitive.
mkldnn_status_t MKLDNN_API mkldnn_deconvolution_backward_weights_desc_init(mkldnn_deconvolution_desc_t *conv_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *diff_weights_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_desc, const mkldnn_dims_t strides, const mkldnn_dims_t padding_l, const mkldnn_dims_t padding_r, mkldnn_padding_kind_t padding_kind)
Initializes a deconvolution descriptor conv_desc for backward propagation with respect to weights usi...
mkldnn_status_t MKLDNN_API mkldnn_rnn_backward_desc_init(mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_rnn_cell_desc_t *rnn_cell_desc, const mkldnn_rnn_direction_t direction, const mkldnn_memory_desc_t *src_layer_desc, const mkldnn_memory_desc_t *src_iter_desc, const mkldnn_memory_desc_t *weights_layer_desc, const mkldnn_memory_desc_t *weights_iter_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_layer_desc, const mkldnn_memory_desc_t *dst_iter_desc, const mkldnn_memory_desc_t *diff_src_layer_desc, const mkldnn_memory_desc_t *diff_src_iter_desc, const mkldnn_memory_desc_t *diff_weights_layer_desc, const mkldnn_memory_desc_t *diff_weights_iter_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_layer, const mkldnn_memory_desc_t *diff_dst_iter_desc)
Initializes a rnn descriptor rnn_desc for backward propagation using prop_kind, rnn_cell_desc, direction, and memory descriptors.
mkldnn_status_t MKLDNN_API mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr_t attr, int count, int mask, const float *scales)
Sets output scales for primitive operations.
mkldnn_status_t MKLDNN_API mkldnn_inner_product_backward_weights_desc_init(mkldnn_inner_product_desc_t *ip_desc, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *diff_weights_desc, const mkldnn_memory_desc_t *diff_bias_desc, const mkldnn_memory_desc_t *diff_dst_desc)
Initializes an inner product descriptor ip_desc for backward propagation with respect to weights usin...
mkldnn_status_t MKLDNN_API mkldnn_rnn_forward_desc_init(mkldnn_rnn_desc_t *rnn_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_rnn_cell_desc_t *rnn_cell_desc, const mkldnn_rnn_direction_t direction, const mkldnn_memory_desc_t *src_layer_desc, const mkldnn_memory_desc_t *src_iter_desc, const mkldnn_memory_desc_t *weights_layer_desc, const mkldnn_memory_desc_t *weights_iter_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_layer_desc, const mkldnn_memory_desc_t *dst_iter_desc)
Initializes a rnn descriptor rnn_desc for forward propagation using prop_kind, rnn_cell_desc, direction, and memory descriptors.
An opaque structure to describe a primitive descriptor iterator .
mkldnn_status_t MKLDNN_API mkldnn_inner_product_forward_desc_init(mkldnn_inner_product_desc_t *ip_desc, mkldnn_prop_kind_t prop_kind, const mkldnn_memory_desc_t *src_desc, const mkldnn_memory_desc_t *weights_desc, const mkldnn_memory_desc_t *bias_desc, const mkldnn_memory_desc_t *dst_desc)
Initializes an inner product descriptor ip_desc for forward propagation using prop_kind (possible val...