1047 #define DNNL_MAX_NDIMS 12 1051 #define DNNL_RUNTIME_DIM_VAL INT64_MIN 1056 #define DNNL_RUNTIME_SIZE_VAL ((size_t)DNNL_RUNTIME_DIM_VAL) 1060 static const union {
1063 } DNNL_RUNTIME_F32_VAL_REP = {0x7fc000d0};
1068 #define DNNL_RUNTIME_F32_VAL (DNNL_RUNTIME_F32_VAL_REP.f) 1071 static const int DNNL_RUNTIME_S32_VAL_REP = INT32_MIN;
1076 #define DNNL_RUNTIME_S32_VAL DNNL_RUNTIME_S32_VAL_REP 1137 #define DNNL_RNN_MAX_N_PARTS 4 1249 #define DNNL_MEMORY_NONE (NULL) 1253 #define DNNL_MEMORY_ALLOCATE ((void *)(size_t)-1) 1311 dnnl_dims_t padding[2];
1462 dnnl_dims_t padding[2];
1500 dnnl_dims_t padding[2];
1881 typedef const struct dnnl_engine *const_dnnl_engine_t;
1997 #define DNNL_ARG_SRC_0 1 1998 #define DNNL_ARG_SRC DNNL_ARG_SRC_0 2001 #define DNNL_ARG_SRC_LAYER DNNL_ARG_SRC_0 2004 #define DNNL_ARG_FROM DNNL_ARG_SRC_0 2009 #define DNNL_ARG_SRC_1 2 2010 #define DNNL_ARG_SRC_ITER DNNL_ARG_SRC_1 2015 #define DNNL_ARG_SRC_2 3 2016 #define DNNL_ARG_SRC_ITER_C DNNL_ARG_SRC_2 2021 #define DNNL_ARG_DST_0 17 2022 #define DNNL_ARG_DST DNNL_ARG_DST_0 2025 #define DNNL_ARG_TO DNNL_ARG_DST_0 2028 #define DNNL_ARG_DST_LAYER DNNL_ARG_DST_0 2032 #define DNNL_ARG_DST_1 18 2033 #define DNNL_ARG_DST_ITER DNNL_ARG_DST_1 2038 #define DNNL_ARG_DST_2 19 2039 #define DNNL_ARG_DST_ITER_C DNNL_ARG_DST_2 2044 #define DNNL_ARG_WEIGHTS_0 33 2045 #define DNNL_ARG_WEIGHTS DNNL_ARG_WEIGHTS_0 2048 #define DNNL_ARG_SCALE_SHIFT DNNL_ARG_WEIGHTS_0 2051 #define DNNL_ARG_WEIGHTS_LAYER DNNL_ARG_WEIGHTS_0 2056 #define DNNL_ARG_WEIGHTS_1 34 2057 #define DNNL_ARG_WEIGHTS_ITER DNNL_ARG_WEIGHTS_1 2062 #define DNNL_ARG_WEIGHTS_2 35 2063 #define DNNL_ARG_WEIGHTS_PEEPHOLE DNNL_ARG_WEIGHTS_2 2068 #define DNNL_ARG_WEIGHTS_3 36 2069 #define DNNL_ARG_WEIGHTS_PROJECTION DNNL_ARG_WEIGHTS_3 2074 #define DNNL_ARG_BIAS 41 2077 #define DNNL_ARG_MEAN 49 2078 #define DNNL_ARG_VARIANCE 50 2083 #define DNNL_ARG_WORKSPACE 64 2084 #define DNNL_ARG_SCRATCHPAD 80 2088 #define DNNL_ARG_DIFF_SRC_0 129 2089 #define DNNL_ARG_DIFF_SRC DNNL_ARG_DIFF_SRC_0 2092 #define DNNL_ARG_DIFF_SRC_LAYER DNNL_ARG_DIFF_SRC_0 2097 #define DNNL_ARG_DIFF_SRC_1 130 2098 #define DNNL_ARG_DIFF_SRC_ITER DNNL_ARG_DIFF_SRC_1 2103 #define DNNL_ARG_DIFF_SRC_2 131 2104 #define DNNL_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_2 2109 #define DNNL_ARG_DIFF_DST_0 145 2110 #define DNNL_ARG_DIFF_DST DNNL_ARG_DIFF_DST_0 2113 #define DNNL_ARG_DIFF_DST_LAYER DNNL_ARG_DIFF_DST_0 2118 #define DNNL_ARG_DIFF_DST_1 146 2119 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 2124 #define DNNL_ARG_DIFF_DST_2 147 2125 #define DNNL_ARG_DIFF_DST_ITER_C DNNL_ARG_DIFF_DST_2 2130 #define DNNL_ARG_DIFF_WEIGHTS_0 161 2131 #define DNNL_ARG_DIFF_WEIGHTS DNNL_ARG_DIFF_WEIGHTS_0 2134 #define DNNL_ARG_DIFF_SCALE_SHIFT DNNL_ARG_DIFF_WEIGHTS_0 2137 #define DNNL_ARG_DIFF_WEIGHTS_LAYER DNNL_ARG_DIFF_WEIGHTS_0 2142 #define DNNL_ARG_DIFF_WEIGHTS_1 162 2143 #define DNNL_ARG_DIFF_WEIGHTS_ITER DNNL_ARG_DIFF_WEIGHTS_1 2148 #define DNNL_ARG_DIFF_WEIGHTS_2 163 2149 #define DNNL_ARG_DIFF_WEIGHTS_PEEPHOLE DNNL_ARG_DIFF_WEIGHTS_2 2154 #define DNNL_ARG_DIFF_WEIGHTS_3 164 2155 #define DNNL_ARG_DIFF_WEIGHTS_PROJECTION DNNL_ARG_DIFF_WEIGHTS_3 2160 #define DNNL_ARG_DIFF_BIAS 169 2163 #define DNNL_ARG_ATTR_OUTPUT_SCALES 513 2167 #define DNNL_ARG_MULTIPLE_SRC 1024 2168 #define DNNL_ARG_MULTIPLE_DST 2048 2173 #define DNNL_ARG_ATTR_ZERO_POINTS 4096 2177 #define DNNL_ARG_ATTR_POST_OP_DW 8192 2180 #define DNNL_ARG_ATTR_MULTIPLE_POST_OP_BASE 16384 2184 #define DNNL_ARG_ATTR_MULTIPLE_POST_OP(idx) \ 2185 (DNNL_ARG_ATTR_MULTIPLE_POST_OP_BASE * ((idx) + 1)) 2323 struct dnnl_stream_attr;
2335 #define DNNL_RUNTIME_NONE 0u 2338 #define DNNL_RUNTIME_SEQ 1u 2341 #define DNNL_RUNTIME_OMP 2u 2344 #define DNNL_RUNTIME_TBB 4u 2347 #define DNNL_RUNTIME_THREADPOOL 8u 2350 #define DNNL_RUNTIME_OCL 256u 2364 #define DNNL_JIT_PROFILE_NONE 0u 2367 #define DNNL_JIT_PROFILE_VTUNE 1u 2370 #define DNNL_JIT_PROFILE_LINUX_PERFMAP 2u 2373 #define DNNL_JIT_PROFILE_LINUX_JITDUMP 4u 2377 #define DNNL_JIT_PROFILE_LINUX_JITDUMP_USE_TSC 8u 2380 #define DNNL_JIT_PROFILE_LINUX_PERF \ 2381 (DNNL_JIT_PROFILE_LINUX_JITDUMP | DNNL_JIT_PROFILE_LINUX_PERFMAP) Definition: dnnl_types.h:609
Reduction using sum.
Definition: dnnl_types.h:971
Definition: dnnl_types.h:600
dnnl_memory_desc_t diff_dst_layer_desc
Destination gradient layer memory descriptor.
Definition: dnnl_types.h:1722
dnnl_dims_t dilation
Pooling dilations for spatial dimensions.
Definition: dnnl_types.h:1504
A layer normalization primitive.
Definition: dnnl_types.h:835
Definition: dnnl_types.h:756
Definition: dnnl_types.h:637
plain 7D tensor
Definition: dnnl_types.h:184
destination grad. memory desc
Definition: dnnl_types.h:2285
Definition: dnnl_types.h:624
Definition: dnnl_types.h:722
An element-wise primitive.
Definition: dnnl_types.h:825
dnnl_alg_kind_t activation_kind
Definition: dnnl_types.h:1740
Reduction using lp norm without final pth-root.
Definition: dnnl_types.h:981
dnnl_memory_desc_t weights_layer_desc
Weights layer memory descriptor.
Definition: dnnl_types.h:1689
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1446
6D tensor blocked by 3rd dimension with block size 4
Definition: dnnl_types.h:362
Definition: dnnl_types.h:367
Definition: dnnl_types.h:377
execution engine
Definition: dnnl_types.h:2234
dnnl_memory_desc_t weights_desc
Weights memory descriptor.
Definition: dnnl_types.h:1627
Definition: dnnl_types.h:632
dnnl_memory_desc_t diff_dst_iter_c_desc
Destination gradient iteration memory descriptor for cell state.
Definition: dnnl_types.h:1726
A batch normalization primitive.
Definition: dnnl_types.h:833
Definition: dnnl_types.h:373
Eltwise: bounded_relu.
Definition: dnnl_types.h:888
Definition: dnnl_types.h:1132
Definition: dnnl_types.h:169
3D tensor blocked by 2nd dimension with block size 16
Definition: dnnl_types.h:229
Definition: dnnl_types.h:89
int r
Definition: dnnl_types.h:1117
Definition: dnnl_types.h:529
CPU engine.
Definition: dnnl_types.h:1868
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1547
Eltwise: ReLU (dst for backward)
Definition: dnnl_types.h:915
Definition: dnnl_types.h:605
Definition: dnnl_types.h:717
destination memory desc
Definition: dnnl_types.h:2284
Direct deconvolution.
Definition: dnnl_types.h:870
Definition: dnnl_types.h:759
Definition: dnnl_types.h:400
Definition: dnnl_types.h:650
struct dnnl_memory * dnnl_memory_t
A memory handle.
Definition: dnnl_types.h:1242
A descriptor for an RNN operation.
Definition: dnnl_types.h:1670
Definition: dnnl_types.h:1940
layer normalization descriptor
Definition: dnnl_types.h:2267
Definition: dnnl_types.h:2241
Definition: dnnl_types.h:645
dnnl_primitive_kind_t
Definition: dnnl_types.h:809
Definition: dnnl_types.h:671
permuted 3D tensor
Definition: dnnl_types.h:201
Definition: dnnl_types.h:311
Eltwise: linear.
Definition: dnnl_types.h:886
Definition: dnnl_types.h:679
dnnl_memory_desc_t bias_desc
Bias memory descriptor.
Definition: dnnl_types.h:1631
dnnl_dim_t local_size
Definition: dnnl_types.h:1529
Internal weights format for 2x3 Winograd.
Definition: dnnl_types.h:1108
dnnl_memory_desc_t data_scaleshift_desc
Definition: dnnl_types.h:1595
Definition: dnnl_types.h:276
Definition: dnnl_types.h:686
permuted 12D tensor
Definition: dnnl_types.h:220
void * dnnl_op_desc_t
A pointer to any of the operation descriptors.
Definition: dnnl_types.h:1263
Definition: dnnl_types.h:230
Definition: dnnl_types.h:724
int n_parts
Definition: dnnl_types.h:1142
Definition: dnnl_types.h:670
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1522
A resampling primitive.
Definition: dnnl_types.h:849
dnnl_format_kind_t format_kind
Memory format kind.
Definition: dnnl_types.h:1222
Definition: dnnl_types.h:770
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1550
Definition: dnnl_types.h:951
Any ISA (excepting those listed as initial support)
Definition: dnnl_types.h:2386
dnnl_dims_t strides
Pooling kernel strides for spatial dimensions.
Definition: dnnl_types.h:1456
int oc
Definition: dnnl_types.h:1120
Definition: dnnl_types.h:1155
4D CNN weights tensor, an alias to dnnl_acdb
Definition: dnnl_types.h:473
Undefined data type, used for empty memory descriptors.
Definition: dnnl_types.h:64
Definition: dnnl_types.h:709
Definition: dnnl_types.h:1130
Definition: dnnl_types.h:385
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1789
dnnl_dim_t group_size
Number of groups.
Definition: dnnl_types.h:1343
Definition: dnnl_types.h:224
dnnl_memory_desc_t bias_desc
Bias memory descriptor.
Definition: dnnl_types.h:1297
Definition: dnnl_types.h:685
Definition: dnnl_types.h:370
5D CNN weights tensor, an alias to dnnl_acdeb
Definition: dnnl_types.h:485
dnnl_convolution_desc_t dnnl_deconvolution_desc_t
A descriptor of a deconvolution operation.
Definition: dnnl_types.h:1322
Definition: dnnl_types.h:289
4D CNN weights tensor (incl. groups), an alias to dnnl_abcd
Definition: dnnl_types.h:490
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1408
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1287
Definition: dnnl_types.h:295
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1490
In-order execution.
Definition: dnnl_types.h:2307
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.
Definition: dnnl_types.h:1313
Definition: dnnl_types.h:278
Definition: dnnl_types.h:996
scratchpad memory desc
Definition: dnnl_types.h:2287
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1811
Definition: dnnl_types.h:317
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1441
dnnl_memory_t memory
Input/output memory.
Definition: dnnl_types.h:2194
const void * const_dnnl_op_desc_t
A pointer to any of the operation descriptors (constant variant).
Definition: dnnl_types.h:1265
Intel Streaming SIMD Extensions 4.1 (Intel SSE4.1)
Definition: dnnl_types.h:2389
Eltwise: clip.
Definition: dnnl_types.h:907
Definition: dnnl_types.h:593
Definition: dnnl_types.h:380
Definition: dnnl_types.h:290
Definition: dnnl_types.h:727
Definition: dnnl_types.h:2422
Definition: dnnl_types.h:663
An opaque structure for primitive descriptor attributes.
Definition: dnnl_types.h:591
Definition: dnnl_types.h:331
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1301
Definition: dnnl_types.h:336
logsoftmax descriptor
Definition: dnnl_types.h:2272
Definition: dnnl_types.h:703
permuted 4D tensor
Definition: dnnl_types.h:198
Definition: dnnl_types.h:307
Definition: dnnl_types.h:594
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1448
Definition: dnnl_types.h:376
float lrn_alpha
LRN alpha parameter.
Definition: dnnl_types.h:1531
dnnl_memory_desc_t diff_weights_desc
Weights gradient memory descriptor.
Definition: dnnl_types.h:1629
dnnl_memory_desc_t dst_iter_c_desc
Destination iter memory descriptor for cell state.
Definition: dnnl_types.h:1699
Primitive iterator passed over last primitive descriptor.
Definition: dnnl_types.h:49
plain 8D tensor
Definition: dnnl_types.h:185
int minor
Minor version.
Definition: dnnl_types.h:2356
Definition: dnnl_types.h:411
float alpha
Definition: dnnl_types.h:1741
Definition: dnnl_types.h:335
2D CNN weights tensor, an alias to dnnl_ab
Definition: dnnl_types.h:457
Definition: dnnl_types.h:743
dnnl_memory_desc_t dst_layer_desc
Destination layer memory descriptor.
Definition: dnnl_types.h:1695
plain 11D tensor
Definition: dnnl_types.h:188
permuted 5D tensor
Definition: dnnl_types.h:199
dnnl_engine_kind_t
Kinds of engines.
Definition: dnnl_types.h:1864
Definition: dnnl_types.h:627
struct dnnl_primitive_desc * dnnl_primitive_desc_t
A primitive descriptor handle.
Definition: dnnl_types.h:1907
Definition: dnnl_types.h:264
Definition: dnnl_types.h:735
Definition: dnnl_types.h:604
Eltwise: square root (dst for backward)
Definition: dnnl_types.h:921
non-standard 16-bit (bfloat16 w/ 7 bit mantissa) floating point.
Definition: dnnl_types.h:68
Definition: dnnl_types.h:857
Definition: dnnl_types.h:375
Undefined primitive.
Definition: dnnl_types.h:811
Definition: dnnl_types.h:602
Definition: dnnl_types.h:401
Definition: dnnl_types.h:622
int ldb
Definition: dnnl_types.h:1144
dnnl_dims_t strides
Convolution strides in each spatial dimension.
Definition: dnnl_types.h:1305
Definition: dnnl_types.h:583
int softmax_axis
The axis along which to perform the softmax.
Definition: dnnl_types.h:1417
Unidirectional execution of RNN primitive from left to right.
Definition: dnnl_types.h:1656
Definition: dnnl_types.h:1935
Definition: dnnl_types.h:758
Definition: dnnl_types.h:738
Definition: dnnl_types.h:699
Definition: dnnl_types.h:251
Definition: dnnl_types.h:655
Definition: dnnl_types.h:676
Definition: dnnl_types.h:409
5D CNN weights tensor (incl. groups), an alias to dnnl_acbde
Definition: dnnl_types.h:498
Definition: dnnl_types.h:714
3D tensor blocked by 2nd dimension with block size 4
Definition: dnnl_types.h:235
Definition: dnnl_types.h:341
Definition: dnnl_types.h:579
Definition: dnnl_types.h:404
dnnl_dims_t dims
Definition: dnnl_types.h:1205
const struct dnnl_memory * const_dnnl_memory_t
A constant memory handle.
Definition: dnnl_types.h:1245
Definition: dnnl_types.h:396
Definition: dnnl_types.h:285
Definition: dnnl_types.h:561
binary descriptor
Definition: dnnl_types.h:2271
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1484
dnnl_memory_desc_t diff_weights_iter_desc
Weights gradient iter memory descriptor.
Definition: dnnl_types.h:1718
Definition: dnnl_types.h:589
Definition: dnnl_types.h:257
permuted 4D tensor
Definition: dnnl_types.h:193
Definition: dnnl_types.h:325
Definition: dnnl_types.h:689
Definition: dnnl_types.h:2291
Description of tensor of packed weights for rnn.
Definition: dnnl_types.h:1140
dnnl_wino_memory_format_t wino_format
Definition: dnnl_types.h:1116
A descriptor of a pooling operation.
Definition: dnnl_types.h:1435
2D CNN activations tensor, an alias to dnnl_ab
Definition: dnnl_types.h:434
plain 2D tensor
Definition: dnnl_types.h:178
4D CNN activations tensor, an alias to dnnl_bcda
Definition: dnnl_types.h:450
Definition: dnnl_types.h:539
Definition: dnnl_types.h:416
permuted 5D tensor
Definition: dnnl_types.h:206
Definition: dnnl_types.h:749
pooling version 2 descriptor
Definition: dnnl_types.h:2275
Undefined memory format kind, used for empty memory descriptors.
Definition: dnnl_types.h:82
dnnl_memory_desc_t bias_desc
Bias memory descriptor.
Definition: dnnl_types.h:1787
Definition: dnnl_types.h:642
Definition: dnnl_types.h:333
Reduction using lp norm without final pth-root.
Definition: dnnl_types.h:983
Definition: dnnl_types.h:390
Definition: dnnl_types.h:1664
32-bit signed integer.
Definition: dnnl_types.h:72
dnnl_memory_extra_desc_t extra
Definition: dnnl_types.h:1234
dnnl_memory_desc_t diff_src_iter_c_desc
Source gradient iter memory descriptor for cell state.
Definition: dnnl_types.h:1714
int n
Definition: dnnl_types.h:1143
Definition: dnnl_types.h:355
int inner_nblks
The number of innermost blocks, e.g. 3 in case of OIhw_4i16o4i_
Definition: dnnl_types.h:1094
Direct convolution.
Definition: dnnl_types.h:864
int major
Major version.
Definition: dnnl_types.h:2355
Definition: dnnl_types.h:701
Definition: dnnl_types.h:427
5D tensor blocked by 1st dimension with block size 8
Definition: dnnl_types.h:304
Definition: dnnl_types.h:406
An opaque structure to describe a primitive descriptor iterator.
pooling descriptor
Definition: dnnl_types.h:2264
Convolution algorithm(either direct or Winograd) is chosen just in time.
Definition: dnnl_types.h:868
Definition: dnnl_types.h:423
Reduction using mean.
Definition: dnnl_types.h:975
Definition: dnnl_types.h:687
Definition: dnnl_types.h:719
Definition: dnnl_types.h:269
Definition: dnnl_types.h:405
float lrn_beta
LRN beta parameter.
Definition: dnnl_types.h:1533
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1623
Definition: dnnl_types.h:680
Definition: dnnl_types.h:241
Unidirectional execution of RNN primitive from right to left.
Definition: dnnl_types.h:1658
Definition: dnnl_types.h:633
A deconvolution primitive.
Definition: dnnl_types.h:823
Definition: dnnl_types.h:681
Definition: dnnl_types.h:629
Definition: dnnl_types.h:726
Definition: dnnl_types.h:408
dnnl_memory_desc_t src_iter_desc
Source iteration memory descriptor for hidden state.
Definition: dnnl_types.h:1685
5D tensor blocked by 2nd dimension with block size 8
Definition: dnnl_types.h:330
8-bit unsigned integer.
Definition: dnnl_types.h:76
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1759
Definition: dnnl_types.h:1778
Definition: dnnl_types.h:646
permuted 6D tensor
Definition: dnnl_types.h:212
3D RNN data tensor in the format (batch, seq_length, input channels).
Definition: dnnl_types.h:509
Definition: dnnl_types.h:718
dnnl_alg_kind_t cell_kind
Definition: dnnl_types.h:1679
const char * hash
Git hash of the sources (may be absent)
Definition: dnnl_types.h:2358
5D CNN weights tensor, an alias to dnnl_cdeba
Definition: dnnl_types.h:483
Definition: dnnl_types.h:598
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1476
4D CNN activations tensor, an alias to dnnl_acdb
Definition: dnnl_types.h:448
dnnl_query_t
Definition: dnnl_types.h:2231
Intel Advanced Vector Extensions (Intel AVX)
Definition: dnnl_types.h:2392
Definition: dnnl_types.h:2403
dnnl_memory_desc_t weights_desc
Weights memory descriptor.
Definition: dnnl_types.h:1785
Backward data propagation.
Definition: dnnl_types.h:800
Definition: dnnl_types.h:256
Definition: dnnl_types.h:710
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1839
Definition: dnnl_types.h:760
dnnl_memory_desc_t stat_desc
Definition: dnnl_types.h:1565
Definition: dnnl_types.h:344
A descriptor of a binary operation.
Definition: dnnl_types.h:1752
Definition: dnnl_types.h:270
Definition: dnnl_types.h:608
source gradient memory desc
Definition: dnnl_types.h:2281
plain 10D tensor
Definition: dnnl_types.h:187
Definition: dnnl_types.h:620
A binary primitive.
Definition: dnnl_types.h:843
Definition: dnnl_types.h:414
Definition: dnnl_types.h:2354
reduction descriptor
Definition: dnnl_types.h:2276
Definition: dnnl_types.h:395
Definition: dnnl_types.h:723
4D tensor blocked by 2nd dimension with block size 32
Definition: dnnl_types.h:261
int arg
An argument index, e.g. DNNL_ARG_SRC.
Definition: dnnl_types.h:2193
permuted 11D tensor
Definition: dnnl_types.h:219
Definition: dnnl_types.h:692
dnnl_memory_desc_t diff_bias_desc
Bias gradient memory descriptor.
Definition: dnnl_types.h:1633
Definition: dnnl_types.h:337
LSTM cell.
Definition: dnnl_types.h:941
Packed weights format used in RNN.
Definition: dnnl_types.h:93
Definition: dnnl_types.h:339
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1479
Definition: dnnl_types.h:704
4D CNN weights tensor, an alias to dnnl_bcda
Definition: dnnl_types.h:475
4D tensor blocked by 2nd dimension with block size 16
Definition: dnnl_types.h:255
permuted 7D tensor
Definition: dnnl_types.h:215
Definition: dnnl_types.h:267
float beta
Definition: dnnl_types.h:1396
A reorder primitive.
Definition: dnnl_types.h:813
Definition: dnnl_types.h:677
dnnl_memory_desc_t weights_desc
Weights memory descriptor.
Definition: dnnl_types.h:1293
A descriptor of a convolution operation.
Definition: dnnl_types.h:1277
Definition: dnnl_types.h:644
2D CNN activations tensor, an alias to dnnl_ba
Definition: dnnl_types.h:436
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.
Definition: dnnl_types.h:1639
softmax descriptor
Definition: dnnl_types.h:2263
permuted 5D tensor
Definition: dnnl_types.h:213
Definition: dnnl_types.h:237
Definition: dnnl_types.h:707
dnnl_dims_t kernel
Pooling kernel spatial dimensions.
Definition: dnnl_types.h:1496
Definition: dnnl_types.h:716
Definition: dnnl_types.h:2407
no query
Definition: dnnl_types.h:2232
Definition: dnnl_types.h:771
Definition: dnnl_types.h:606
Definition: dnnl_types.h:421
dnnl_scratchpad_mode_t
Scratchpad mode.
Definition: dnnl_types.h:1918
Definition: dnnl_types.h:1035
Definition: dnnl_types.h:294
Definition: dnnl_types.h:275
batch normalization descriptor
Definition: dnnl_types.h:2266
dnnl_dims_t padded_dims
Size of the data including padding in each dimension.
Definition: dnnl_types.h:1211
A reduction primitive.
Definition: dnnl_types.h:853
Definition: dnnl_types.h:345
Definition: dnnl_types.h:596
dnnl_dim_t offset0
Definition: dnnl_types.h:1219
Definition: dnnl_types.h:403
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1803
Definition: dnnl_types.h:747
5D CNN weights tensor, an alias to dnnl_abcde
Definition: dnnl_types.h:479
Definition: dnnl_types.h:748
Definition: dnnl_types.h:613
Definition: dnnl_types.h:296
float beta
Definition: dnnl_types.h:1742
dnnl_dims_t padded_offsets
Definition: dnnl_types.h:1215
Undefined memory format, used for empty memory descriptors.
Definition: dnnl_types.h:1105
Definition: dnnl_types.h:398
runtime estimation (seconds)
Definition: dnnl_types.h:2240
Backward propagation (with respect to all parameters).
Definition: dnnl_types.h:798
An unspecified engine.
Definition: dnnl_types.h:1866
5D tensor blocked by 1st dimension with block size 16
Definition: dnnl_types.h:302
Eltwise: ReLU.
Definition: dnnl_types.h:874
GPU engine.
Definition: dnnl_types.h:1870
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1763
Definition: dnnl_types.h:326
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1755
Definition: dnnl_types.h:512
Definition: dnnl_types.h:581
Definition: dnnl_types.h:519
6D CNN weights tensor (incl. groups), an alias to dnnl_acbdef
Definition: dnnl_types.h:502
3D tensor blocked by 2nd dimension with block size 32
Definition: dnnl_types.h:233
Definition: dnnl_types.h:711
Definition: dnnl_types.h:2246
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1676
Definition: dnnl_types.h:595
Eltwise: pow.
Definition: dnnl_types.h:909
Definition: dnnl_types.h:343
5D tensor blocked by 2nd dimension with block size 4
Definition: dnnl_types.h:315
dnnl_memory_extra_flags_t
Flags for memory special features.
Definition: dnnl_types.h:1154
Definition: dnnl_types.h:526
Definition: dnnl_types.h:580
permuted 4D tensor
Definition: dnnl_types.h:208
An opaque structure to describe an engine.
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
Definition: dnnl_types.h:1524
Definition: dnnl_types.h:350
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1783
Definition: dnnl_types.h:706
Definition: dnnl_types.h:388
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1289
Definition: dnnl_types.h:372
6D CNN weights tensor (incl. groups), an alias to dnnl_abcdef
Definition: dnnl_types.h:500
dnnl_memory_desc_t diff_data_scaleshift_desc
Definition: dnnl_types.h:1596
Definition: dnnl_types.h:647
Definition: dnnl_types.h:729
Definition: dnnl_types.h:248
Definition: dnnl_types.h:259
Definition: dnnl_types.h:792
int alpha
Definition: dnnl_types.h:1118
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
Definition: dnnl_types.h:1585
Undefined RNN flags.
Definition: dnnl_types.h:1650
A sum primitive.
Definition: dnnl_types.h:819
unsigned flags
Definition: dnnl_types.h:1568
unsigned cpu_runtime
CPU runtime.
Definition: dnnl_types.h:2359
Definition: dnnl_types.h:737
5D CNN weights tensor (incl. groups), an alias to dnnl_decab
Definition: dnnl_types.h:496
Definition: dnnl_types.h:728
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1358
Definition: dnnl_types.h:253
Definition: dnnl_types.h:265
Definition: dnnl_types.h:2399
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
Definition: dnnl_types.h:1413
dnnl_memory_desc_t diff_data_desc
Source and destination gradient memory descriptor.
Definition: dnnl_types.h:1587
A descriptor of a element-wise operation.
Definition: dnnl_types.h:1352
dnnl_dims_t inner_blks
The size of the blocks, e.g. {4, 16, 4} in case of OIhw_4i16o4i
Definition: dnnl_types.h:1096
Definition: dnnl_types.h:721
A descriptor of a Softmax operation.
Definition: dnnl_types.h:1405
Definition: dnnl_types.h:252
An opaque structure for a chain of post operations.
A descriptor of an inner product operation.
Definition: dnnl_types.h:1614
Eltwise: round.
Definition: dnnl_types.h:913
Definition: dnnl_types.h:338
The operation failed because of incorrect function arguments.
Definition: dnnl_types.h:45
Definition: dnnl_types.h:238
Definition: dnnl_types.h:740
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.
Definition: dnnl_types.h:1502
Definition: dnnl_types.h:614
Definition: dnnl_types.h:413
Definition: dnnl_types.h:360
3D RNN data tensor in the format (seq_length, batch, input channels).
Definition: dnnl_types.h:507
dnnl_memory_desc_t diff_weights_projection_desc
Definition: dnnl_types.h:1734
dnnl_rnn_direction_t
A direction of RNN primitive execution.
Definition: dnnl_types.h:1654
Internal weights format for 2x3 Winograd.
Definition: dnnl_types.h:1107
inner product descriptor
Definition: dnnl_types.h:2268
Definition: dnnl_types.h:392
Definition: dnnl_types.h:582
5D tensor blocked by 2nd dimension with block size 16
Definition: dnnl_types.h:306
A descriptor of reduction operation.
Definition: dnnl_types.h:1828
6D tensor blocked by 2nd dimension with block size 4
Definition: dnnl_types.h:364
Definition: dnnl_types.h:720
dnnl_status_t
Status values returned by the library functions.
Definition: dnnl_types.h:39
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1452
Reduction using lp norm.
Definition: dnnl_types.h:979
Binary mul.
Definition: dnnl_types.h:955
dnnl_memory_desc_t dst_iter_desc
Destination iter memory descriptor for hidden state.
Definition: dnnl_types.h:1697
int ic_block
Definition: dnnl_types.h:1121
Definition: dnnl_types.h:236
A softmax primitive.
Definition: dnnl_types.h:827
Definition: dnnl_types.h:242
const struct dnnl_primitive_desc * const_dnnl_primitive_desc_t
A constant primitive descriptor handle.
Definition: dnnl_types.h:1910
Definition: dnnl_types.h:739
int64_t dnnl_dim_t
A type to describe tensor dimension.
Definition: dnnl_types.h:1079
Forward data propagation (alias for dnnl_forward_training).
Definition: dnnl_types.h:796
Definition: dnnl_types.h:769
primitive kind
Definition: dnnl_types.h:2235
matrix multiplication (matmul) descriptor
Definition: dnnl_types.h:2273
Default stream configuration.
Definition: dnnl_types.h:2311
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1635
dnnl_memory_desc_t stat_desc
Definition: dnnl_types.h:1602
const struct dnnl_primitive * const_dnnl_primitive_t
A constant primitive handle.
Definition: dnnl_types.h:1994
LRN within a single channel.
Definition: dnnl_types.h:937
dnnl_data_type_t
Data type specification.
Definition: dnnl_types.h:62
Definition: dnnl_types.h:399
plain 4D tensor
Definition: dnnl_types.h:180
Definition: dnnl_types.h:1087
Definition: dnnl_types.h:250
plain 6D tensor
Definition: dnnl_types.h:183
Definition: dnnl_types.h:323
Eltwise: exp (dst for backward)
Definition: dnnl_types.h:925
size_t offset_compensation
Definition: dnnl_types.h:1148
Definition: dnnl_types.h:599
Winograd convolution.
Definition: dnnl_types.h:866
5D tensor blocked by 3rd dimension with block size 4
Definition: dnnl_types.h:352
Definition: dnnl_types.h:669
Definition: dnnl_types.h:742
Definition: dnnl_types.h:741
dnnl_wino_desc_t wino_desc
Tensor of weights for integer 8bit winograd convolution.
Definition: dnnl_types.h:1228
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1371
Definition: dnnl_types.h:365
Definition: dnnl_types.h:415
dnnl_format_tag_t
Definition: dnnl_types.h:164
Definition: dnnl_types.h:532
Max pooling.
Definition: dnnl_types.h:927
Eltwise: natural logarithm.
Definition: dnnl_types.h:905
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1815
Definition: dnnl_types.h:616
Definition: dnnl_types.h:381
float batch_norm_epsilon
Batch normalization epsilon parameter.
Definition: dnnl_types.h:1567
Definition: dnnl_types.h:273
Definition: dnnl_types.h:635
Definition: dnnl_types.h:349
float p
Definition: dnnl_types.h:1853
Definition: dnnl_types.h:641
Definition: dnnl_types.h:402
dnnl_memory_desc_t weights_projection_desc
Definition: dnnl_types.h:1707
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1583
plain 12D tensor
Definition: dnnl_types.h:189
Description of tensor of weights for winograd 2x3 convolution.
Definition: dnnl_types.h:1115
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1411
Definition: dnnl_types.h:564
Definition: dnnl_types.h:708
Reduction using lp norm.
Definition: dnnl_types.h:977
Definition: dnnl_types.h:665
dnnl_memory_desc_t bias_desc
Bias memory descriptor.
Definition: dnnl_types.h:1693
3D CNN weights tensor, an alias to dnnl_cba
Definition: dnnl_types.h:465
Binary div.
Definition: dnnl_types.h:961
4D tensor blocked by 2nd dimension with block size 8
Definition: dnnl_types.h:282
Out-of-order execution.
Definition: dnnl_types.h:2309
Binary min.
Definition: dnnl_types.h:959
dnnl_memory_desc_t dst_desc
Destination memory descriptor.
Definition: dnnl_types.h:1841
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1516
Definition: dnnl_types.h:652
Definition: dnnl_types.h:334
Definition: dnnl_types.h:664
dnnl_format_kind_t
Memory format kind.
Definition: dnnl_types.h:80
permuted 3D tensor
Definition: dnnl_types.h:195
4D tensor blocked by 3rd dimension with block size 4
Definition: dnnl_types.h:300
Definition: dnnl_types.h:700
Definition: dnnl_types.h:552
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1673
Average pooling (alias for dnnl_pooling_avg_exclude_padding)
Definition: dnnl_types.h:933
Definition: dnnl_types.h:247
dnnl_memory_desc_t diff_src_layer_desc
Source gradient layer memory descriptor.
Definition: dnnl_types.h:1710
Definition: dnnl_types.h:246
int axis
Axis for shuffling.
Definition: dnnl_types.h:1341
Definition: dnnl_types.h:225
dnnl_rnn_packed_desc_t rnn_packed_desc
Tensor of packed weights for RNN.
Definition: dnnl_types.h:1230
3D tensor blocked by 2nd dimension with block size 8
Definition: dnnl_types.h:245
Binary add.
Definition: dnnl_types.h:953
3D CNN activations tensor, an alias to dnnl_abc
Definition: dnnl_types.h:442
deconvolution descriptor
Definition: dnnl_types.h:2260
A pooling primitive.
Definition: dnnl_types.h:829
Definition: dnnl_types.h:610
dnnl_dims_t strides
Definition: dnnl_types.h:1090
Definition: dnnl_types.h:286
rnn descriptor
Definition: dnnl_types.h:2269
Definition: dnnl_types.h:280
Definition: dnnl_types.h:634
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1280
Definition: dnnl_types.h:668
Definition: dnnl_types.h:607
Eltwise: logistic.
Definition: dnnl_types.h:892
dnnl_memory_desc_t diff_data_scaleshift_desc
Definition: dnnl_types.h:1561
A descriptor of a shuffle operation.
Definition: dnnl_types.h:1330
Definition: dnnl_types.h:277
const struct dnnl_primitive_desc_iterator * const_dnnl_primitive_desc_iterator_t
A constant primitive descriptor iterator handle.
Definition: dnnl_types.h:1899
dnnl_dims_t kernel
Pooling kernel spatial dimensions.
Definition: dnnl_types.h:1458
Definition: dnnl_types.h:631
int oc_block
Definition: dnnl_types.h:1122
Eltwise: hyperbolic tangent non-linearity (tanh) (dst for backward)
Definition: dnnl_types.h:917
Definition: dnnl_types.h:675
Winograd deconvolution.
Definition: dnnl_types.h:872
Definition: dnnl_types.h:733
permuted 4D tensor
Definition: dnnl_types.h:209
Definition: dnnl_types.h:683
number of outputs expected
Definition: dnnl_types.h:2238
Definition: dnnl_types.h:320
Definition: dnnl_types.h:308
Definition: dnnl_types.h:266
#define DNNL_MAX_NDIMS
Definition: dnnl_types.h:1047
Definition: dnnl_types.h:319
Definition: dnnl_types.h:730
5D CNN weights tensor (incl. groups), an alias to dnnl_abcde
Definition: dnnl_types.h:494
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
Definition: dnnl_types.h:1625
float lrn_k
LRN k parameter.
Definition: dnnl_types.h:1535
Definition: dnnl_types.h:324
Definition: dnnl_types.h:640
dnnl_rnn_direction_t direction
The direction of RNN primitive execution.
Definition: dnnl_types.h:1681
Definition: dnnl_types.h:693
6D CNN weights tensor (incl. groups), an alias to dnnl_defcab
Definition: dnnl_types.h:504
GEMM descriptor (internal)
Definition: dnnl_types.h:2270
plain 1D tensor
Definition: dnnl_types.h:177
Definition: dnnl_types.h:1661
Reduction using min.
Definition: dnnl_types.h:969
permuted 2D tensor
Definition: dnnl_types.h:200
permuted 5D tensor
Definition: dnnl_types.h:211
Definition: dnnl_types.h:384
permuted 6D tensor
Definition: dnnl_types.h:197
dnnl_memory_desc_t src_iter_c_desc
Source iteration memory descriptor for cell state.
Definition: dnnl_types.h:1687
stub
Definition: dnnl_types.h:2257
dnnl_memory_desc_t diff_desc
Source and Destination of gradient memory descriptor.
Definition: dnnl_types.h:1415
3D CNN weights tensor, an alias to dnnl_bca
Definition: dnnl_types.h:467
Definition: dnnl_types.h:555
propagation kind
Definition: dnnl_types.h:2254
plain 9D tensor
Definition: dnnl_types.h:186
An inner product primitive.
Definition: dnnl_types.h:837
Definition: dnnl_types.h:258
Definition: dnnl_types.h:698
Definition: dnnl_types.h:1009
Definition: dnnl_types.h:751
Definition: dnnl_types.h:374
Definition: dnnl_types.h:656
Definition: dnnl_types.h:750
Definition: dnnl_types.h:546
GRU cell.
Definition: dnnl_types.h:943
Definition: dnnl_types.h:309
3D CNN activations tensor, an alias to dnnl_acb
Definition: dnnl_types.h:444
The operation was successful.
Definition: dnnl_types.h:41
Definition: dnnl_types.h:597
Definition: dnnl_types.h:585
Definition: dnnl_types.h:648
A descriptor of a Layer Normalization operation.
Definition: dnnl_types.h:1577
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1333
8-bit signed integer.
Definition: dnnl_types.h:74
convolution descriptor
Definition: dnnl_types.h:2259
Definition: dnnl_types.h:394
dnnl_memory_desc_t weights_peephole_desc
Definition: dnnl_types.h:1703
RNN cell.
Definition: dnnl_types.h:939
float adj_scale
Definition: dnnl_types.h:1125
Alias for dnnl_unidirectional_left2right.
Definition: dnnl_types.h:1666
A (out-of-place) concat primitive.
Definition: dnnl_types.h:817
Definition: dnnl_types.h:322
Definition: dnnl_types.h:657
Internal weights format for 2x3 Winograd.
Definition: dnnl_types.h:1109
Definition: dnnl_types.h:617
Definition: dnnl_types.h:422
dnnl_dims_t dilates
Convolution dilates in each spatial dimension.
Definition: dnnl_types.h:1307
const struct dnnl_stream * const_dnnl_stream_t
A constant execution stream handle.
Definition: dnnl_types.h:2320
2D RNN statistics tensor, an alias to dnnl_ba
Definition: dnnl_types.h:440
5D CNN activations tensor, an alias to dnnl_acdeb
Definition: dnnl_types.h:454
Definition: dnnl_types.h:2412
Undefined memory format tag.
Definition: dnnl_types.h:166
Reduction using mul.
Definition: dnnl_types.h:973
permuted 5D tensor
Definition: dnnl_types.h:196
Eltwise: square root.
Definition: dnnl_types.h:884
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
Definition: dnnl_types.h:1552
Definition: dnnl_types.h:682
Definition: dnnl_types.h:590
Definition: dnnl_types.h:757
Definition: dnnl_types.h:736
Definition: dnnl_types.h:279
int patch
Patch version.
Definition: dnnl_types.h:2357
Definition: dnnl_types.h:713
Definition: dnnl_types.h:767
Definition: dnnl_types.h:391
Definition: dnnl_types.h:862
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1837
Definition: dnnl_types.h:558
permuted 3D tensor
Definition: dnnl_types.h:204
Definition: dnnl_types.h:243
permuted 8D tensor
Definition: dnnl_types.h:216
Definition: dnnl_types.h:702
dnnl_memory_desc_t diff_data_desc
Source and destination gradient memory descriptor.
Definition: dnnl_types.h:1526
Definition: dnnl_types.h:321
Definition: dnnl_types.h:651
source memory desc
Definition: dnnl_types.h:2280
Eltwise: swish.
Definition: dnnl_types.h:903
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1519
#define DNNL_RNN_MAX_N_PARTS
Definition: dnnl_types.h:1137
Definition: dnnl_types.h:674
Definition: dnnl_types.h:588
Definition: dnnl_types.h:348
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1831
Definition: dnnl_types.h:310
Definition: dnnl_types.h:356
Definition: dnnl_types.h:283
struct dnnl_stream_attr * dnnl_stream_attr_t
An execution stream attributes handle.
Definition: dnnl_types.h:2325
Definition: dnnl_types.h:1188
Definition: dnnl_types.h:274
dnnl_wino_memory_format_t
Winograd-specific formats.
Definition: dnnl_types.h:1103
2D RNN statistics tensor, an alias to dnnl_ab
Definition: dnnl_types.h:438
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1580
A descriptor of a pooling operation.
Definition: dnnl_types.h:1473
dnnl_data_type_t data_type
Data type of the tensor elements.
Definition: dnnl_types.h:1208
Definition: dnnl_types.h:297
A matrix multiplication primitive.
Definition: dnnl_types.h:847
Queried element is not required for given primitive.
Definition: dnnl_types.h:53
Definition: dnnl_types.h:410
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.
Definition: dnnl_types.h:1464
Eltwise: erf-based gelu.
Definition: dnnl_types.h:911
permuted 9D tensor
Definition: dnnl_types.h:217
Definition: dnnl_types.h:1163
dnnl_blocking_desc_t blocking
Definition: dnnl_types.h:1226
dnnl_dims_t inner_idxs
Definition: dnnl_types.h:1099
Definition: dnnl_types.h:653
Definition: dnnl_types.h:625
The operation failed due to an out-of-memory condition.
Definition: dnnl_types.h:43
Backward weights propagation.
Definition: dnnl_types.h:802
size_t size
Definition: dnnl_types.h:1149
5D tensor blocked by 2nd dimension with block size 32
Definition: dnnl_types.h:313
3D CNN weights tensor, an alias to dnnl_acb
Definition: dnnl_types.h:463
Definition: dnnl_types.h:764
Definition: dnnl_types.h:318
dnnl_memory_desc_t diff_weights_desc
Weights gradient memory descriptor.
Definition: dnnl_types.h:1295
Definition: dnnl_types.h:298
Definition: dnnl_types.h:2305
dnnl_memory_desc_t data_scaleshift_desc
Definition: dnnl_types.h:1560
workspace memory desc
Definition: dnnl_types.h:2286
Definition: dnnl_types.h:615
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1781
Definition: dnnl_types.h:378
Definition: dnnl_types.h:383
Definition: dnnl_types.h:762
Definition: dnnl_types.h:659
size_t size
Definition: dnnl_types.h:1126
Definition: dnnl_types.h:660
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
Definition: dnnl_types.h:1303
eltwise descriptor
Definition: dnnl_types.h:2262
Definition: dnnl_types.h:667
number of inputs expected
Definition: dnnl_types.h:2237
const struct dnnl_stream_attr * const_dnnl_stream_attr_t
A constant execution stream attributes handle.
Definition: dnnl_types.h:2327
shuffle descriptor
Definition: dnnl_types.h:2261
Definition: dnnl_types.h:412
Average pooling include padding.
Definition: dnnl_types.h:929
dnnl_memory_desc_t diff_data_desc
Source and destination gradient memory descriptor.
Definition: dnnl_types.h:1554
Weights format used in 8bit Winograd convolution.
Definition: dnnl_types.h:91
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
Definition: dnnl_types.h:1813
32-bit/single-precision floating point.
Definition: dnnl_types.h:70
permuted 5D tensor
Definition: dnnl_types.h:210
Definition: dnnl_types.h:2192
Definition: dnnl_types.h:2417
Definition: dnnl_types.h:628
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1806
Definition: dnnl_types.h:567
struct dnnl_primitive_attr * dnnl_primitive_attr_t
A primitive descriptor attributes handle that controls primitive behavior.
Definition: dnnl_types.h:1952
Definition: dnnl_types.h:393
lrn descriptor
Definition: dnnl_types.h:2265
dnnl_memory_desc_t src_layer_desc
Source layer memory descriptor.
Definition: dnnl_types.h:1683
Definition: dnnl_types.h:240
Definition: dnnl_types.h:592
Definition: dnnl_types.h:761
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
Definition: dnnl_types.h:1817
4D tensor blocked by 1st and 2nd dimension with block size 8
Definition: dnnl_types.h:288
Definition: dnnl_types.h:327
A shuffle primitive.
Definition: dnnl_types.h:815
for creating scratchpad memory
Definition: dnnl_types.h:2249
Definition: dnnl_types.h:387
Definition: dnnl_types.h:226
float layer_norm_epsilon
Layer normalization epsilon parameter.
Definition: dnnl_types.h:1604
dnnl_memory_desc_t diff_weights_peephole_desc
Definition: dnnl_types.h:1730
struct dnnl_engine * dnnl_engine_t
An engine handle.
Definition: dnnl_types.h:1877
Definition: dnnl_types.h:619
Definition: dnnl_types.h:643
Definition: dnnl_types.h:691
unsigned int flags
RNN cell flags.
Definition: dnnl_types.h:1737
permuted 4D tensor
Definition: dnnl_types.h:205
Definition: dnnl_types.h:386
4D tensor blocked by 2nd dimension with block size 4
Definition: dnnl_types.h:263
Binary max.
Definition: dnnl_types.h:957
Definition: dnnl_types.h:768
Definition: dnnl_types.h:666
Definition: dnnl_types.h:85
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1336
Definition: dnnl_types.h:239
Definition: dnnl_types.h:1165
2D CNN weights tensor, an alias to dnnl_ba
Definition: dnnl_types.h:459
Average pooling exclude padding.
Definition: dnnl_types.h:931
Definition: dnnl_types.h:1131
Definition: dnnl_types.h:658
Definition: dnnl_types.h:573
dnnl_alg_kind_t
Kinds of algorithms.
Definition: dnnl_types.h:861
4D CNN weights tensor, an alias to dnnl_cdba
Definition: dnnl_types.h:471
Definition: dnnl_types.h:766
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
Definition: dnnl_types.h:1492
Reduction using max.
Definition: dnnl_types.h:967
5D CNN weights tensor, an alias to dnnl_bacde
Definition: dnnl_types.h:481
permuted 3D tensor
Definition: dnnl_types.h:207
dnnl_rnn_flags_t
Flags for RNN cell.
Definition: dnnl_types.h:1648
dnnl_memory_desc_t diff_dst_iter_desc
Destination gradient iteration memory descriptor for hidden state.
Definition: dnnl_types.h:1724
dnnl_memory_desc_t src_desc
Source memory descriptor.
Definition: dnnl_types.h:1486
A pooling version 2 primitive (pooling with dilation support).
Definition: dnnl_types.h:851
int ic2_block
Definition: dnnl_types.h:1123
4D CNN weights tensor (incl. groups), an alias to dnnl_dcab
Definition: dnnl_types.h:492
Definition: dnnl_types.h:231
Definition: dnnl_types.h:705
Definition: dnnl_types.h:752
Definition: dnnl_types.h:332
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1621
Definition: dnnl_types.h:765
Definition: dnnl_types.h:397
An LRN primitive.
Definition: dnnl_types.h:831
Definition: dnnl_types.h:346
Definition: dnnl_types.h:420
A descriptor of a Local Response Normalization (LRN) operation.
Definition: dnnl_types.h:1513
Definition: dnnl_types.h:284
Definition: dnnl_types.h:271
dnnl_softmax_desc_t dnnl_logsoftmax_desc_t
Definition: dnnl_types.h:1427
int ndims
Number of dimensions.
Definition: dnnl_types.h:1190
dnnl_stream_flags_t
Stream flags.
Definition: dnnl_types.h:2302
dnnl_dim_t dnnl_dims_t[DNNL_MAX_NDIMS]
A type to describe tensor dimensions.
Definition: dnnl_types.h:1082
Undefined propagation type.
Definition: dnnl_types.h:785
dnnl_memory_desc_t diff_data_desc
Source and destination gradient memory descriptor.
Definition: dnnl_types.h:1375
Definition: dnnl_types.h:696
Primitive or engine failed on execution.
Definition: dnnl_types.h:51
Definition: dnnl_types.h:368
Definition: dnnl_types.h:684
op descriptor
Definition: dnnl_types.h:2258
6D tensor blocked by 2nd dimension with block size 8
Definition: dnnl_types.h:359
int ic
Definition: dnnl_types.h:1119
Definition: dnnl_types.h:601
dnnl_memory_desc_t weights_iter_desc
Weights iteration memory descriptor.
Definition: dnnl_types.h:1691
const struct dnnl_post_ops * const_dnnl_post_ops_t
A constant post operation chain handle.
Definition: dnnl_types.h:1981
Definition: dnnl_types.h:715
Eltwise: exponent.
Definition: dnnl_types.h:894
Definition: dnnl_types.h:419
1D tensor, an alias to dnnl_a
Definition: dnnl_types.h:432
Local response normalization (LRN) across multiple channels.
Definition: dnnl_types.h:935
Definition: dnnl_types.h:227
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1438
resampling descriptor
Definition: dnnl_types.h:2274
const struct dnnl_primitive_attr * const_dnnl_primitive_attr_t
A constant primitive descriptor attributes handle.
Definition: dnnl_types.h:1955
Definition: dnnl_types.h:1166
Definition: dnnl_types.h:697
A matrix multiplication primitive (internal).
Definition: dnnl_types.h:841
Definition: dnnl_types.h:342
Definition: dnnl_types.h:291
dnnl_memory_desc_t diff_bias_desc
Bias gradient memory descriptor.
Definition: dnnl_types.h:1720
Definition: dnnl_types.h:366
plain 3D tensor
Definition: dnnl_types.h:179
Definition: dnnl_types.h:1022
Definition: dnnl_types.h:570
A descriptor of a Batch Normalization operation.
Definition: dnnl_types.h:1544
5D CNN weights tensor, an alias to dnnl_bcdea
Definition: dnnl_types.h:487
struct dnnl_primitive * dnnl_primitive_t
A primitive handle.
Definition: dnnl_types.h:1992
Definition: dnnl_types.h:672
Intel Advanced Vector Extensions 2 (Intel AVX2)
Definition: dnnl_types.h:2395
Definition: dnnl_types.h:576
struct dnnl_stream * dnnl_stream_t
An execution stream handle.
Definition: dnnl_types.h:2318
4D CNN weights tensor, an alias to dnnl_bacd
Definition: dnnl_types.h:477
Eltwise: exponential linear unit (elu) (dst for backward)
Definition: dnnl_types.h:919
Eltwise: exponential linear unit (elu)
Definition: dnnl_types.h:878
Forward data propagation (alias for dnnl_forward_inference).
Definition: dnnl_types.h:794
6D tensor blocked by 2nd dimension with block size 16
Definition: dnnl_types.h:354
Definition: dnnl_types.h:649
Definition: dnnl_types.h:746
Definition: dnnl_types.h:549
Definition: dnnl_types.h:379
An opaque structure to describe a primitive descriptor.
dnnl_prop_kind_t prop_kind
Definition: dnnl_types.h:1284
struct dnnl_primitive_desc_iterator * dnnl_primitive_desc_iterator_t
A primitive descriptor iterator handle.
Definition: dnnl_types.h:1896
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
Definition: dnnl_types.h:1488
Eltwise: abs.
Definition: dnnl_types.h:882
Definition: dnnl_types.h:673
dnnl_rnn_packed_memory_format_t format
Definition: dnnl_types.h:1141
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
Definition: dnnl_types.h:1373
Internal weights format for 4x3 Winograd.
Definition: dnnl_types.h:1111
Definition: dnnl_types.h:788
Definition: dnnl_types.h:371
Definition: dnnl_types.h:734
permuted 5D tensor
Definition: dnnl_types.h:194
dnnl_memory_desc_t data_desc
Definition: dnnl_types.h:1339
Eltwise: tanh-based gelu (alias for dnnl_eltwise_gelu_tanh)
Definition: dnnl_types.h:901
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
Definition: dnnl_types.h:1291
Definition: dnnl_types.h:755
A rnn primitive.
Definition: dnnl_types.h:839
4D CNN activations tensor, an alias to dnnl_abcd
Definition: dnnl_types.h:446
A logsoftmax primitive.
Definition: dnnl_types.h:845
Definition: dnnl_types.h:357
permuted 5D tensor
Definition: dnnl_types.h:203
The operation failed because requested functionality is not implemented.
Definition: dnnl_types.h:47
Definition: dnnl_types.h:899
dnnl_data_type_t accum_data_type
The accumulator data type. Initialized automatically.
Definition: dnnl_types.h:1791
Definition: dnnl_types.h:418
dnnl_memory_desc_t diff_src_iter_desc
Source gradient iter memory descriptor for hidden state.
Definition: dnnl_types.h:1712
dnnl_memory_desc_t diff_src_desc
Source gradient memory descriptor.
Definition: dnnl_types.h:1450
Eltwise: hyperbolic tangent non-linearity (tanh)
Definition: dnnl_types.h:876
Definition: dnnl_types.h:712
struct dnnl_post_ops * dnnl_post_ops_t
A post operation chain handle.
Definition: dnnl_types.h:1978
permuted 10D tensor
Definition: dnnl_types.h:218
A descriptor of resampling operation.
Definition: dnnl_types.h:1800
Definition: dnnl_types.h:249
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
Definition: dnnl_types.h:1454
Definition: dnnl_types.h:623
Definition: dnnl_types.h:603
Definition: dnnl_types.h:584
dnnl_prop_kind_t
Kinds of propagation.
Definition: dnnl_types.h:782
Definition: dnnl_types.h:1164
permuted 6D tensor
Definition: dnnl_types.h:214
stub
Definition: dnnl_types.h:2279
int oc2_block
Definition: dnnl_types.h:1124
Definition: dnnl_types.h:292
Definition: dnnl_types.h:389
Nearest Neighbor Resampling Method.
Definition: dnnl_types.h:963
permuted 4D tensor
Definition: dnnl_types.h:202
Definition: dnnl_types.h:753
dnnl_memory_desc_t diff_dst_desc
Destination gradient memory descriptor.
Definition: dnnl_types.h:1637
Definition: dnnl_types.h:732
plain 4D tensor
Definition: dnnl_types.h:181
Definition: dnnl_types.h:630
Definition: dnnl_types.h:621
Definition: dnnl_types.h:369
16-bit/half-precision floating point.
Definition: dnnl_types.h:66
Definition: dnnl_types.h:690
Definition: dnnl_types.h:754
dnnl_rnn_packed_memory_format_t
Definition: dnnl_types.h:1129
source engine
Definition: dnnl_types.h:2251
dnnl_memory_desc_t diff_bias_desc
Bias gradient memory descriptor.
Definition: dnnl_types.h:1299
dnnl_alg_kind_t alg_kind
Definition: dnnl_types.h:1809
Definition: dnnl_types.h:618
Definition: dnnl_types.h:328
Definition: dnnl_types.h:293
Definition: dnnl_types.h:347
Definition: dnnl_types.h:316
dnnl_normalization_flags_t
Flags for normalization primitives.
Definition: dnnl_types.h:987
weights grad. memory desc
Definition: dnnl_types.h:2283
Definition: dnnl_types.h:417
Definition: dnnl_types.h:763
A convolution primitive.
Definition: dnnl_types.h:821
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1355
memory desc of an execute argument
Definition: dnnl_types.h:2288
Definition: dnnl_types.h:678
Backward bias propagation.
Definition: dnnl_types.h:804
5D CNN activations tensor, an alias to dnnl_abcde
Definition: dnnl_types.h:452
4D CNN weights tensor, an alias to dnnl_abcd
Definition: dnnl_types.h:469
Eltwise: logistic (dst for backward)
Definition: dnnl_types.h:923
Definition: dnnl_types.h:340
dnnl_memory_desc_t diff_weights_layer_desc
Weights gradient layer memory descriptor.
Definition: dnnl_types.h:1716
weights memory descriptor desc
Definition: dnnl_types.h:2282
unsigned flags
Definition: dnnl_types.h:1605
Definition: dnnl_types.h:272
Definition: dnnl_types.h:407
Definition: dnnl_types.h:654
unsigned gpu_runtime
GPU runtime.
Definition: dnnl_types.h:2360
Linear Resampling Method.
Definition: dnnl_types.h:965
3D CNN weights tensor, an alias to dnnl_abc
Definition: dnnl_types.h:461
dnnl_cpu_isa_t
CPU instruction set flags.
Definition: dnnl_types.h:2384
Eltwise: soft_relu.
Definition: dnnl_types.h:890
plain 5D tensor
Definition: dnnl_types.h:182
dnnl_dims_t strides
Pooling kernel strides for spatial dimensions.
Definition: dnnl_types.h:1494
Definition: dnnl_types.h:688
destination engine
Definition: dnnl_types.h:2252
dnnl_primitive_kind_t primitive_kind
Definition: dnnl_types.h:1617
Definition: dnnl_types.h:382
Definition: dnnl_types.h:626
Eltwise: square.
Definition: dnnl_types.h:880
Definition: dnnl_types.h:636
Definition: dnnl_types.h:268