Memory

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_memory_extra_desc_t
	Description of extra information stored in memory. More...
struct	mkldnn_memory_desc_t
	Memory descriptor. More...
struct	mkldnn_memory
	An opaque structure to describe a memory. More...

Macros
#define	MKLDNN_MAX_NDIMS   12
	Maximum number of dimensions a tensor can have. More...
#define	MKLDNN_RNN_MAX_N_PARTS   4
	Maximum number of parts of RNN weights tensor that require separate computation. More...

Typedefs
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.
typedef const struct mkldnn_memory *	const_mkldnn_memory_t
	A constant memory handle.

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_memory_extra_flags_t { , mkldnn_memory_extra_flag_compensation_conv_s8s8 = 0x1U }
	Flags for memory special features. More...

Detailed Description

Macro Definition Documentation

MKLDNN_MAX_NDIMS

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

MKLDNN_RNN_MAX_N_PARTS

#define MKLDNN_RNN_MAX_N_PARTS   4

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

Enumeration Type Documentation

mkldnn_wino_memory_format_t

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.

mkldnn_memory_extra_flags_t

enum mkldnn_memory_extra_flags_t

Flags for memory special features.

Enumerator
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) }