mxnet
Modules | Classes | Macros | Typedefs | Enumerations
Auxiliary types for memory description
Collaboration diagram for Auxiliary types for memory description:

Modules

 Operation descriptors
 

Classes

struct  mkldnn_blocking_desc_t
 
struct  mkldnn_wino_desc_t
 
struct  mkldnn_rnn_packed_desc_t
 
struct  mkldnn_convolution_desc_t
 
struct  mkldnn_shuffle_desc_t
 
struct  mkldnn_eltwise_desc_t
 
struct  mkldnn_softmax_desc_t
 
struct  mkldnn_pooling_desc_t
 
struct  mkldnn_lrn_desc_t
 
struct  mkldnn_batch_normalization_desc_t
 
struct  mkldnn_inner_product_desc_t
 
struct  mkldnn_rnn_cell_desc_t
 
struct  mkldnn_rnn_desc_t
 

Macros

#define TENSOR_MAX_DIMS   12
 
#define MKLDNN_RNN_MAX_N_PARTS   4
 

Typedefs

typedef int mkldnn_dims_t[TENSOR_MAX_DIMS]
 
typedef ptrdiff_t mkldnn_strides_t[TENSOR_MAX_DIMS]
 
typedef mkldnn_convolution_desc_t mkldnn_deconvolution_desc_t
 

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 }
 
enum  mkldnn_rnn_direction_t {
  mkldnn_unidirectional_left2right, mkldnn_unidirectional_right2left, mkldnn_bidirectional_concat, mkldnn_bidirectional_sum,
  mkldnn_unidirectional = mkldnn_unidirectional_left2right
}
 

Detailed Description

Macro Definition Documentation

#define MKLDNN_RNN_MAX_N_PARTS   4
#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.

Typedef Documentation

A descriptor of a deconvolution operation.

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.

Enumeration Type Documentation

Flags for RNN cell.

Enumerator
mkldnn_rnn_cell_with_relu 
mkldnn_rnn_cell_with_clipping 

A direction of RNN primitive execution.

Enumerator
mkldnn_unidirectional_left2right 
mkldnn_unidirectional_right2left 
mkldnn_bidirectional_concat 
mkldnn_bidirectional_sum 
mkldnn_unidirectional 
Enumerator
mkldnn_packed_format_undef 
mkldnn_ldigo_p 
mkldnn_ldgoi_p 
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.