Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.21.0
Performance library for Deep Learning
|
Modules | |
Operation descriptors | |
Classes | |
struct | mkldnn_blocking_desc_t |
Generic description of blocked data layout for most memory formats. More... | |
struct | mkldnn_wino_desc_t |
Description of tensor of weights for winograd 2x3 convolution. More... | |
struct | mkldnn_rnn_packed_desc_t |
Description of tensor of packed weights for rnn. More... | |
struct | mkldnn_convolution_desc_t |
A descriptor of a convolution operation. More... | |
struct | mkldnn_shuffle_desc_t |
A descriptor of a shuffle operation. More... | |
struct | mkldnn_eltwise_desc_t |
A descriptor of a element-wise operation. More... | |
struct | mkldnn_softmax_desc_t |
A descriptor of a Softmax operation. More... | |
struct | mkldnn_pooling_desc_t |
A descriptor of a pooling operation. More... | |
struct | mkldnn_lrn_desc_t |
A descriptor of a Local Response Normalization (LRN) operation. More... | |
struct | mkldnn_batch_normalization_desc_t |
A descriptor of a Batch Normalization operation. More... | |
struct | mkldnn_inner_product_desc_t |
A descriptor of an inner product operation. More... | |
struct | mkldnn_rnn_cell_desc_t |
struct | mkldnn_rnn_desc_t |
A descriptor for an RNN operation. More... | |
Macros | |
#define | TENSOR_MAX_DIMS 12 |
Maximum number of dimensions a tensor can have. More... | |
#define | MKLDNN_RNN_MAX_N_PARTS 4 |
Typedefs | |
typedef int | mkldnn_dims_t[TENSOR_MAX_DIMS] |
A type to describe tensor dimensions. More... | |
typedef ptrdiff_t | mkldnn_strides_t[TENSOR_MAX_DIMS] |
A type to describe strides within a tensor. More... | |
typedef mkldnn_convolution_desc_t | mkldnn_deconvolution_desc_t |
A descriptor of a deconvolution operation. More... | |
Enumerations | |
enum | mkldnn_wino_memory_format_t { mkldnn_wino_undef = 0, mkldnn_wino_wei_aaOIoi, mkldnn_wino_wei_aaOio, mkldnn_wino_wei_aaOBiOo, mkldnn_wino_wei_OBaaIBOIio } |
enum | mkldnn_rnn_packed_memory_format_t { mkldnn_packed_format_undef = 0, mkldnn_ldigo_p, mkldnn_ldgoi_p } |
enum | mkldnn_rnn_cell_flags_t { mkldnn_rnn_cell_with_relu = 0x1U, mkldnn_rnn_cell_with_clipping = 0x2U } |
Flags for RNN cell. More... | |
enum | mkldnn_rnn_direction_t { mkldnn_unidirectional_left2right, mkldnn_unidirectional_right2left, mkldnn_bidirectional_concat, mkldnn_bidirectional_sum, mkldnn_unidirectional = mkldnn_unidirectional_left2right } |
A direction of RNN primitive execution. More... | |
#define TENSOR_MAX_DIMS 12 |
Maximum number of dimensions a tensor can have.
Only restricts the amount of space used for the tensor description. Individual computational primitives may support only tensors of certain dimensions.
#define MKLDNN_RNN_MAX_N_PARTS 4 |
typedef int mkldnn_dims_t[TENSOR_MAX_DIMS] |
A type to describe tensor dimensions.
typedef ptrdiff_t mkldnn_strides_t[TENSOR_MAX_DIMS] |
A type to describe strides within a tensor.
A descriptor of a deconvolution operation.