Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.10
Performance library for Deep Learning
|
Memory primitive that describes the data. More...
#include <mkldnn.hpp>
Classes | |
struct | desc |
A memory descriptor. More... | |
struct | primitive_desc |
A memory primitive descriptor. More... | |
Public Types | |
enum | data_type { data_undef = mkldnn_data_type_undef, f32 = mkldnn_f32, s32 = mkldnn_s32, s16 = mkldnn_s16, s8 = mkldnn_s8, u8 = mkldnn_u8 } |
Data type specification. See mkldnn_data_type_t for a detailed description. More... | |
enum | format { format_undef = mkldnn_format_undef, any = mkldnn_any, blocked = mkldnn_blocked, x = mkldnn_x, nc = mkldnn_nc, nchw = mkldnn_nchw, nhwc = mkldnn_nhwc, chwn = mkldnn_chwn, nChw8c = mkldnn_nChw8c, nChw16c = mkldnn_nChw16c, oi = mkldnn_oi, io = mkldnn_io, oihw = mkldnn_oihw, ihwo = mkldnn_ihwo, hwio = mkldnn_hwio, oIhw8i = mkldnn_oIhw8i, oIhw16i = mkldnn_oIhw16i, OIhw8i8o = mkldnn_OIhw8i8o, OIhw16i16o = mkldnn_OIhw16i16o, OIhw8o8i = mkldnn_OIhw8o8i, OIhw16o16i = mkldnn_OIhw16o16i, OIhw8i16o2i = mkldnn_OIhw8i16o2i, OIhw8o16i2o = mkldnn_OIhw8o16i2o, Oihw8o = mkldnn_Oihw8o, Oihw16o = mkldnn_Oihw16o, Ohwi8o = mkldnn_Ohwi8o, Ohwi16o = mkldnn_Ohwi16o, OhIw16o4i = mkldnn_OhIw16o4i, goihw = mkldnn_goihw, gOIhw8i8o = mkldnn_gOIhw8i8o, gOIhw16i16o = mkldnn_gOIhw16i16o, gOIhw8i16o2i = mkldnn_gOIhw8i16o2i, gOIhw8o16i2o = mkldnn_gOIhw8o16i2o, gOihw8o = mkldnn_gOihw8o, gOihw16o = mkldnn_gOihw16o, gOhwi8o = mkldnn_gOhwi8o, gOhwi16o = mkldnn_gOhwi16o, gOIhw8o8i = mkldnn_gOIhw8o8i, gOIhw16o16i = mkldnn_gOIhw16o16i, gOhIw16o4i = mkldnn_gOhIw16o4i } |
Memory format specification. See mkldnn_memory_format_t for a detailed description. More... | |
typedef std::vector< std::remove_extent< mkldnn_dims_t >::type > | dims |
Public Member Functions | |
memory (const primitive &aprimitive) | |
Constructs a memory primitive from a generic primitive. More... | |
memory (const primitive_desc &adesc) | |
Constructs a memory primitive. More... | |
memory (const primitive_desc &adesc, void *ahandle) | |
primitive_desc | get_primitive_desc () const |
Returns the descriptor of the memory primitive. More... | |
void * | get_data_handle () const |
Returns a handle of the data contained in the memory primitive. On the CPU engine, this is a pointer to the allocated memory. More... | |
void | set_data_handle (void *handle) const |
![]() | |
const_mkldnn_primitive_desc_t | get_primitive_desc () const |
Returns the descriptor of the underlying C API primitive. More... | |
![]() | |
handle (const handle &other) | |
handle & | operator= (const handle &other) |
bool | operator== (const handle &other) const |
bool | operator!= (const handle &other) const |
void | reset (mkldnn_primitive_t t, bool weak=false) |
Resets the value of a C handle. More... | |
mkldnn_primitive_t | get () const |
Returns the value of the underlying C handle. More... | |
Static Public Member Functions | |
template<typename T > | |
static void | validate_dims (std::vector< T > v) |
static mkldnn_data_type_t | convert_to_c (data_type adata_type) |
static mkldnn_memory_format_t | convert_to_c (format aformat) |
Additional Inherited Members | |
![]() | |
handle (mkldnn_primitive_t t=0, bool weak=false) | |
Constructs a C handle wrapper. More... | |
bool | operator== (const mkldnn_primitive_t other) const |
bool | operator!= (const mkldnn_primitive_t other) const |
Memory primitive that describes the data.