|
| #define | DNNL_MAX_NDIMS 12 |
| | Maximum number of dimensions a tensor can have.
|
|
#define | DNNL_RUNTIME_DIM_VAL INT64_MIN |
| | A wildcard value for dimensions that are unknown at a primitive creation time.
|
| #define | DNNL_RUNTIME_SIZE_VAL ((size_t)DNNL_RUNTIME_DIM_VAL) |
| | A size_t counterpart of the DNNL_RUNTIME_DIM_VAL.
|
|
#define | DNNL_RUNTIME_F32_VAL (DNNL_RUNTIME_F32_VAL_REP.f) |
| | A wildcard value for floating point values that are unknown at a primitive creation time.
|
|
#define | DNNL_RUNTIME_S32_VAL DNNL_RUNTIME_S32_VAL_REP |
| | A wildcard value for int32_t values that are unknown at a primitive creation time.
|
|
#define | DNNL_RNN_MAX_N_PARTS 4 |
| | Maximum number of parts of RNN weights tensor that require separate computation.
|
|
#define | DNNL_ARG_SRC_0 1 |
| | Source argument #0.
|
| #define | DNNL_ARG_SRC DNNL_ARG_SRC_0 |
| | A special mnemonic for source argument for primitives that have a single source.
|
| #define | DNNL_ARG_SRC_LAYER DNNL_ARG_SRC_0 |
| | A special mnemonic for RNN input vector.
|
| #define | DNNL_ARG_FROM DNNL_ARG_SRC_0 |
| | A special mnemonic for reorder source argument.
|
|
#define | DNNL_ARG_SRC_1 2 |
| | Source argument #1.
|
| #define | DNNL_ARG_SRC_ITER DNNL_ARG_SRC_1 |
| | A special mnemonic for RNN input recurrent hidden state vector.
|
|
#define | DNNL_ARG_SRC_2 3 |
| | Source argument #2.
|
| #define | DNNL_ARG_SRC_ITER_C DNNL_ARG_SRC_2 |
| | A special mnemonic for RNN input recurrent cell state vector.
|
|
#define | DNNL_ARG_DST_0 17 |
| | Destination argument #0.
|
| #define | DNNL_ARG_DST DNNL_ARG_DST_0 |
| | A special mnemonic for destination argument for primitives that have a single destination.
|
| #define | DNNL_ARG_TO DNNL_ARG_DST_0 |
| | A special mnemonic for reorder destination argument.
|
|
#define | DNNL_ARG_DST_LAYER DNNL_ARG_DST_0 |
| | A special mnemonic for RNN output vector. An alias for DNNL_ARG_DST_0.
|
|
#define | DNNL_ARG_DST_1 18 |
| | Destination argument #1.
|
| #define | DNNL_ARG_DST_ITER DNNL_ARG_DST_1 |
| | A special mnemonic for RNN input recurrent hidden state vector.
|
|
#define | DNNL_ARG_DST_2 19 |
| | Destination argument #2.
|
| #define | DNNL_ARG_DST_ITER_C DNNL_ARG_DST_2 |
| | A special mnemonic for LSTM output recurrent cell state vector.
|
|
#define | DNNL_ARG_WEIGHTS_0 33 |
| | Weights argument #0.
|
| #define | DNNL_ARG_WEIGHTS DNNL_ARG_WEIGHTS_0 |
| | A special mnemonic for primitives that have a single weights argument.
|
| #define | DNNL_ARG_SCALE_SHIFT DNNL_ARG_WEIGHTS_0 |
| | A special mnemonic for scale and shift argument of normalization primitives.
|
| #define | DNNL_ARG_WEIGHTS_LAYER DNNL_ARG_WEIGHTS_0 |
| | A special mnemonic for RNN weights applied to the layer input.
|
|
#define | DNNL_ARG_WEIGHTS_1 34 |
| | Weights argument #1.
|
| #define | DNNL_ARG_WEIGHTS_ITER DNNL_ARG_WEIGHTS_1 |
| | A special mnemonic for RNN weights applied to the recurrent input.
|
|
#define | DNNL_ARG_WEIGHTS_2 35 |
| | Weights argument #2.
|
| #define | DNNL_ARG_WEIGHTS_PEEPHOLE DNNL_ARG_WEIGHTS_2 |
| | A special mnemonic for RNN weights applied to the peephole weights.
|
|
#define | DNNL_ARG_WEIGHTS_3 36 |
| | Weights argument #3.
|
| #define | DNNL_ARG_WEIGHTS_PROJECTION DNNL_ARG_WEIGHTS_3 |
| | A special mnemonic for RNN weights applied to the projection weights.
|
|
#define | DNNL_ARG_BIAS 41 |
| | Bias tensor argument.
|
|
#define | DNNL_ARG_MEAN 49 |
| | Mean values tensor argument.
|
|
#define | DNNL_ARG_VARIANCE 50 |
| | Variance values tensor argument.
|
| #define | DNNL_ARG_WORKSPACE 64 |
| | Workspace tensor argument.
|
|
#define | DNNL_ARG_SCRATCHPAD 80 |
| | Scratchpad (temporary storage) tensor argument.
|
|
#define | DNNL_ARG_DIFF_SRC_0 129 |
| | Gradient (diff) of the source argument #0.
|
| #define | DNNL_ARG_DIFF_SRC DNNL_ARG_DIFF_SRC_0 |
| | A special mnemonic for primitives that have a single diff source argument.
|
| #define | DNNL_ARG_DIFF_SRC_LAYER DNNL_ARG_DIFF_SRC_0 |
| | A special mnemonic for gradient (diff) of RNN input vector.
|
|
#define | DNNL_ARG_DIFF_SRC_1 130 |
| | Gradient (diff) of the source argument #1.
|
| #define | DNNL_ARG_DIFF_SRC_ITER DNNL_ARG_DIFF_SRC_1 |
| | A special mnemonic for gradient (diff) of RNN input recurrent hidden state vector.
|
|
#define | DNNL_ARG_DIFF_SRC_2 131 |
| | Gradient (diff) of the source argument #2.
|
| #define | DNNL_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_2 |
| | A special mnemonic for gradient (diff) of RNN input recurrent cell state vector.
|
|
#define | DNNL_ARG_DIFF_DST_0 145 |
| | Gradient (diff) of the destination argument #0.
|
| #define | DNNL_ARG_DIFF_DST DNNL_ARG_DIFF_DST_0 |
| | A special mnemonic for primitives that have a single diff destination argument.
|
| #define | DNNL_ARG_DIFF_DST_LAYER DNNL_ARG_DIFF_DST_0 |
| | A special mnemonic for gradient (diff) of RNN output vector.
|
|
#define | DNNL_ARG_DIFF_DST_1 146 |
| | Gradient (diff) of the destination argument #1.
|
| #define | DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 |
| | A special mnemonic for gradient (diff) of RNN input recurrent hidden state vector.
|
|
#define | DNNL_ARG_DIFF_DST_2 147 |
| | Gradient (diff) of the destination argument #2.
|
| #define | DNNL_ARG_DIFF_DST_ITER_C DNNL_ARG_DIFF_DST_2 |
| | A special mnemonic for gradient (diff) of RNN input recurrent cell state vector.
|
|
#define | DNNL_ARG_DIFF_WEIGHTS_0 161 |
| | Gradient (diff) of the weights argument #0.
|
| #define | DNNL_ARG_DIFF_WEIGHTS DNNL_ARG_DIFF_WEIGHTS_0 |
| | A special mnemonic for primitives that have a single diff weights argument.
|
| #define | DNNL_ARG_DIFF_SCALE_SHIFT DNNL_ARG_DIFF_WEIGHTS_0 |
| | A special mnemonic for diff of scale and shift argument of normalization primitives.
|
| #define | DNNL_ARG_DIFF_WEIGHTS_LAYER DNNL_ARG_DIFF_WEIGHTS_0 |
| | A special mnemonic for diff of RNN weights applied to the layer input.
|
|
#define | DNNL_ARG_DIFF_WEIGHTS_1 162 |
| | Gradient (diff) of the weights argument #1.
|
| #define | DNNL_ARG_DIFF_WEIGHTS_ITER DNNL_ARG_DIFF_WEIGHTS_1 |
| | A special mnemonic for diff of RNN weights applied to the recurrent input.
|
|
#define | DNNL_ARG_DIFF_WEIGHTS_2 163 |
| | Gradient (diff) of the weights argument #2.
|
| #define | DNNL_ARG_DIFF_WEIGHTS_PEEPHOLE DNNL_ARG_DIFF_WEIGHTS_2 |
| | A special mnemonic for diff of RNN weights applied to the peephole weights.
|
|
#define | DNNL_ARG_DIFF_WEIGHTS_3 164 |
| | Gradient (diff) of the weights argument #3.
|
| #define | DNNL_ARG_DIFF_WEIGHTS_PROJECTION DNNL_ARG_DIFF_WEIGHTS_3 |
| | A special mnemonic for diff of RNN weights applied to the projection weights.
|
|
#define | DNNL_ARG_DIFF_BIAS 169 |
| | Gradient (diff) of the bias tensor argument.
|
|
#define | DNNL_ARG_ATTR_OUTPUT_SCALES 513 |
| | Output scaling factors provided at execution time.
|
|
#define | DNNL_ARG_MULTIPLE_SRC 1024 |
| | Starting index for source arguments for primitives that take a variable number of source arguments.
|
|
#define | DNNL_ARG_MULTIPLE_DST 2048 |
| | Starting index for destination arguments for primitives that produce a variable number of destination arguments.
|
|
#define | DNNL_ARG_ATTR_ZERO_POINTS 4096 |
| | Zero points provided at execution time.
|
| #define | DNNL_ARG_ATTR_POST_OP_DW 8192 |
| | Arguments for fused depthwise convolution.
|
|
#define | DNNL_RUNTIME_NONE 0u |
| | No runtime (disabled)
|
|
#define | DNNL_RUNTIME_SEQ 1u |
| | Sequential runtime (CPU only)
|
|
#define | DNNL_RUNTIME_OMP 2u |
| | OpenMP runtime (CPU only)
|
|
#define | DNNL_RUNTIME_TBB 4u |
| | TBB runtime (CPU only)
|
|
#define | DNNL_RUNTIME_THREADPOOL 8u |
| | Threadpool runtime (CPU only)
|
|
#define | DNNL_RUNTIME_OCL 256u |
| | OpenCL runtime.
|
|
#define | DNNL_JIT_PROFILE_NONE 0u |
| | Disable profiling completely.
|
|
#define | DNNL_JIT_PROFILE_VTUNE 1u |
| | Enable VTune Amplifier integration.
|
|
#define | DNNL_JIT_PROFILE_LINUX_PERFMAP 2u |
| | Enable Linux perf integration via perfmap files.
|
|
#define | DNNL_JIT_PROFILE_LINUX_JITDUMP 4u |
| | Enable Linux perf integration via jitdump files.
|
| #define | DNNL_JIT_PROFILE_LINUX_JITDUMP_USE_TSC 8u |
| | Instruct Linux perf integration via jitdump files to use TSC.
|
|
#define | DNNL_JIT_PROFILE_LINUX_PERF (DNNL_JIT_PROFILE_LINUX_JITDUMP | DNNL_JIT_PROFILE_LINUX_PERFMAP) |
| | Enable Linux perf integration (both jitdump and perfmap)
|
|
|
typedef int64_t | dnnl_dim_t |
| | A type to describe tensor dimension.
|
|
typedef dnnl_dim_t | dnnl_dims_t[DNNL_MAX_NDIMS] |
| | A type to describe tensor dimensions.
|
|
typedef struct dnnl_memory * | dnnl_memory_t |
| | A memory handle.
|
|
typedef const struct dnnl_memory * | const_dnnl_memory_t |
| | A constant memory handle.
|
|
typedef void * | dnnl_op_desc_t |
| | A pointer to any of the operation descriptors.
|
|
typedef const void * | const_dnnl_op_desc_t |
| | A pointer to any of the operation descriptors (constant variant).
|
|
typedef dnnl_convolution_desc_t | dnnl_deconvolution_desc_t |
| | A descriptor of a deconvolution operation.
|
| typedef dnnl_softmax_desc_t | dnnl_logsoftmax_desc_t |
| | A descriptor of a LogSoftmax operation.
|
|
typedef struct dnnl_engine * | dnnl_engine_t |
| | An engine handle.
|
|
typedef struct dnnl_primitive_desc_iterator * | dnnl_primitive_desc_iterator_t |
| | A primitive descriptor iterator handle.
|
|
typedef const struct dnnl_primitive_desc_iterator * | const_dnnl_primitive_desc_iterator_t |
| | A constant primitive descriptor iterator handle.
|
|
typedef struct dnnl_primitive_desc * | dnnl_primitive_desc_t |
| | A primitive descriptor handle.
|
|
typedef const struct dnnl_primitive_desc * | const_dnnl_primitive_desc_t |
| | A constant primitive descriptor handle.
|
|
typedef struct dnnl_primitive_attr * | dnnl_primitive_attr_t |
| | A primitive descriptor attributes handle that controls primitive behavior.
|
|
typedef const struct dnnl_primitive_attr * | const_dnnl_primitive_attr_t |
| | A constant primitive descriptor attributes handle.
|
|
typedef struct dnnl_post_ops * | dnnl_post_ops_t |
| | A post operation chain handle.
|
|
typedef const struct dnnl_post_ops * | const_dnnl_post_ops_t |
| | A constant post operation chain handle.
|
|
typedef struct dnnl_primitive * | dnnl_primitive_t |
| | A primitive handle.
|
|
typedef const struct dnnl_primitive * | const_dnnl_primitive_t |
| | A constant primitive handle.
|
|
typedef struct dnnl_stream * | dnnl_stream_t |
| | An execution stream handle.
|
|
typedef const struct dnnl_stream * | const_dnnl_stream_t |
| | A constant execution stream handle.
|
|
typedef struct dnnl_stream_attr * | dnnl_stream_attr_t |
| | An execution stream attributes handle.
|
|
typedef const struct dnnl_stream_attr * | const_dnnl_stream_attr_t |
| | A constant execution stream attributes handle.
|