Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  0.21.0
Performance library for Deep Learning
Modules | Classes | Macros | Typedefs | Enumerations
Auxiliary types for memory description

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...
 

Detailed Description

Macro Definition Documentation

◆ TENSOR_MAX_DIMS

#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.

◆ MKLDNN_RNN_MAX_N_PARTS

#define MKLDNN_RNN_MAX_N_PARTS   4

Typedef Documentation

◆ mkldnn_dims_t

typedef int mkldnn_dims_t[TENSOR_MAX_DIMS]

A type to describe tensor dimensions.

◆ mkldnn_strides_t

typedef ptrdiff_t mkldnn_strides_t[TENSOR_MAX_DIMS]

A type to describe strides within a tensor.

◆ mkldnn_deconvolution_desc_t

A descriptor of a deconvolution operation.

Enumeration Type Documentation

◆ mkldnn_wino_memory_format_t

Enumerator
mkldnn_wino_undef 

Undefined memory format, used for empty memory descriptors.

mkldnn_wino_wei_aaOIoi 

Tensors of weights for 2x3 winograd convolutions.

mkldnn_wino_wei_aaOio 
mkldnn_wino_wei_aaOBiOo 
mkldnn_wino_wei_OBaaIBOIio 

Tensor of weights for 4x3 convolution.

◆ mkldnn_rnn_packed_memory_format_t

Enumerator
mkldnn_packed_format_undef 
mkldnn_ldigo_p 
mkldnn_ldgoi_p 

◆ mkldnn_rnn_cell_flags_t

Flags for RNN cell.

Enumerator
mkldnn_rnn_cell_with_relu 
mkldnn_rnn_cell_with_clipping 

◆ mkldnn_rnn_direction_t

A direction of RNN primitive execution.

Enumerator
mkldnn_unidirectional_left2right 
mkldnn_unidirectional_right2left 
mkldnn_bidirectional_concat 
mkldnn_bidirectional_sum 
mkldnn_unidirectional