mkldnn_memory_desc_t Struct Reference

#include <mkldnn_types.h>

Collaboration diagram for mkldnn_memory_desc_t:

Public Attributes
int	ndims
	Number of dimensions. More...
mkldnn_dims_t	dims
mkldnn_data_type_t	data_type
	Data type of the tensor elements. More...
mkldnn_dims_t	padded_dims
	Size of the data including padding in each dimension. More...
mkldnn_dims_t	padded_offsets
mkldnn_dim_t	offset0
mkldnn_format_kind_t	format_kind
	Memory format kind. More...
union {
mkldnn_blocking_desc_t   blocking
mkldnn_wino_desc_t   wino_desc
	Tensor of weights for integer 8bit winograd convolution. More...
mkldnn_rnn_packed_desc_t   rnn_packed_desc
	Tensor of packed weights for RNN. More...
}	format_desc
mkldnn_memory_extra_desc_t	extra

Detailed Description

Memory descriptor. The description is based on a number of dimensions, dimensions themselves, plus information about elements type and memory format. Additionally, contains format-specific descriptions of the data layout.

Member Data Documentation

mkldnn_blocking_desc_t mkldnn_memory_desc_t::blocking

Description of the data layout for memory formats that use blocking.

mkldnn_data_type_t mkldnn_memory_desc_t::data_type

Data type of the tensor elements.

mkldnn_dims_t mkldnn_memory_desc_t::dims

Dimensions in the following order:

• CNN data tensors: mini-batch, channel, spatial ({N, C, [[D,] H,] W})
• CNN weight tensors: group (optional), output channel, input channel, spatial ({[G,] O, I, [[D,] H,] W})
• RNN data tensors: time, mini-batch, channels ({T, N, C}) or layers, directions, states, mini-batch, channels ({L, D, S, N, C})
• RNN weight tensor: layers, directions, input channel, gates, output channels ({L, D, I, G, O}).

Note

The order of dimensions does not depend on the memory format, so whether the data is laid out in mkldnn_nchw or mkldnn_nhwc the dims for 4D CN data tensor would be {N, C, H, W}.

mkldnn_memory_extra_desc_t mkldnn_memory_desc_t::extra

union { ... } mkldnn_memory_desc_t::format_desc

mkldnn_format_kind_t mkldnn_memory_desc_t::format_kind

Memory format kind.

int mkldnn_memory_desc_t::ndims

Number of dimensions.

mkldnn_dim_t mkldnn_memory_desc_t::offset0

Offset from memory origin to the current block, non-zero only in a description of a memory sub-block.

mkldnn_dims_t mkldnn_memory_desc_t::padded_dims

Size of the data including padding in each dimension.

mkldnn_dims_t mkldnn_memory_desc_t::padded_offsets

Per-dimension offset from the padding to actual data, the top-level tensor with offsets applied must lie within the padding area.

mkldnn_rnn_packed_desc_t mkldnn_memory_desc_t::rnn_packed_desc

Tensor of packed weights for RNN.

mkldnn_wino_desc_t mkldnn_memory_desc_t::wino_desc

Tensor of weights for integer 8bit winograd convolution.

---

The documentation for this struct was generated from the following file:

• /work/mxnet/include/mkldnn/mkldnn_types.h