Memory

Collaboration diagram for Memory:

Classes
struct	mkldnn_blocking_desc_t
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_memory_extra_desc_t
	Description of extra information stored in memory. More...
struct	mkldnn_memory_desc_t
struct	mkldnn_memory

Macros
#define	MKLDNN_MAX_NDIMS   12
#define	MKLDNN_RNN_MAX_N_PARTS   4
#define	MKLDNN_MEMORY_NONE   (NULL)
#define	MKLDNN_MEMORY_ALLOCATE   ((void *)(size_t)-1)

Typedefs
typedef int64_t	mkldnn_dim_t
	A type to describe tensor dimension. More...
typedef mkldnn_dim_t	mkldnn_dims_t[MKLDNN_MAX_NDIMS]
	A type to describe tensor dimensions. More...
typedef struct mkldnn_memory *	mkldnn_memory_t
	A memory handle. More...
typedef const struct mkldnn_memory *	const_mkldnn_memory_t
	A constant memory handle. 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 }
	Winograd-specific formats. More...
enum	mkldnn_rnn_packed_memory_format_t { mkldnn_packed_format_undef = 0, mkldnn_ldigo_p, mkldnn_ldgoi_p }
enum	mkldnn_memory_extra_flags_t { mkldnn_memory_extra_flag_none = 0x0U, mkldnn_memory_extra_flag_compensation_conv_s8s8 = 0x1U, mkldnn_memory_extra_flag_scale_adjust = 0x2U }
	Flags for memory special features. More...

Detailed Description

Macro Definition Documentation

#define MKLDNN_MAX_NDIMS   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_MEMORY_ALLOCATE   ((void *)(size_t)-1)

#define MKLDNN_MEMORY_NONE   (NULL)

#define MKLDNN_RNN_MAX_N_PARTS   4

Maximum number of parts of RNN weights tensor that require separate computation.

Typedef Documentation

typedef const struct mkldnn_memory* const_mkldnn_memory_t

A constant memory handle.

typedef int64_t mkldnn_dim_t

A type to describe tensor dimension.

typedef mkldnn_dim_t mkldnn_dims_t[MKLDNN_MAX_NDIMS]

A type to describe tensor dimensions.

typedef struct mkldnn_memory* mkldnn_memory_t

A memory handle.

Enumeration Type Documentation

enum mkldnn_memory_extra_flags_t

Flags for memory special features.

Enumerator
mkldnn_memory_extra_flag_none
mkldnn_memory_extra_flag_compensation_conv_s8s8	Indicates the weights have an additional buffer, that depends on the compensation_mask. For instance, in 4D case with the compensation mask equals (1 << 0) the additional buffer would consist of OC values: O[oc : 0,OC] = -128 * SUM(ic : 0,IC; kh : 0,KH; kw : 0,KW){ weights(oc, ic, kh, kw) }
mkldnn_memory_extra_flag_scale_adjust

enum mkldnn_rnn_packed_memory_format_t

Enumerator
mkldnn_packed_format_undef
mkldnn_ldigo_p
mkldnn_ldgoi_p

enum mkldnn_wino_memory_format_t

Winograd-specific formats.

Enumerator
mkldnn_wino_undef	Undefined memory format, used for empty memory descriptors.
mkldnn_wino_wei_aaOIoi	Internal weights format for 2x3 Winograd.
mkldnn_wino_wei_aaOio	Internal weights format for 2x3 Winograd.
mkldnn_wino_wei_aaOBiOo	Internal weights format for 2x3 Winograd.
mkldnn_wino_wei_OBaaIBOIio	Internal weights format for 4x3 Winograd.