Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  0.10
Performance library for Deep Learning
Classes | Typedefs | Enumerations | Enumerator | Functions | Variables | Friends
Memory

Classes

struct  mkldnn::primitive::at
 A wrapper structure to specify a particular output of a primitive. More...
 
struct  mkldnn::handle_traits< mkldnn_primitive_desc_t >
 
struct  mkldnn::memory::desc
 A memory descriptor. More...
 
struct  mkldnn::memory::primitive_desc
 A memory primitive descriptor. More...
 
struct  mkldnn::memory
 Memory primitive that describes the data. More...
 
struct  mkldnn::reorder::primitive_desc
 
struct  mkldnn::reorder
 
struct  mkldnn::view::primitive_desc
 
struct  mkldnn::view
 
struct  mkldnn::concat::primitive_desc
 
struct  mkldnn::concat
 
struct  mkldnn::sum::primitive_desc
 
struct  mkldnn::sum
 
struct  mkldnn::stream
 
struct  mkldnn::convolution_forward::desc
 
struct  mkldnn::convolution_forward::primitive_desc
 
struct  mkldnn::convolution_forward
 
struct  mkldnn::convolution_backward_data::desc
 
struct  mkldnn::convolution_backward_data::primitive_desc
 
struct  mkldnn::convolution_backward_data
 
struct  mkldnn::convolution_backward_weights::desc
 
struct  mkldnn::convolution_backward_weights::primitive_desc
 
struct  mkldnn::convolution_backward_weights
 
struct  mkldnn::convolution_relu_forward::desc
 
struct  mkldnn::convolution_relu_forward::primitive_desc
 
struct  mkldnn::convolution_relu_forward
 
struct  mkldnn::lrn_forward::desc
 
struct  mkldnn::lrn_forward::primitive_desc
 
struct  mkldnn::lrn_forward
 
struct  mkldnn::lrn_backward::desc
 
struct  mkldnn::lrn_backward::primitive_desc
 
struct  mkldnn::lrn_backward
 
struct  mkldnn::pooling_forward::desc
 
struct  mkldnn::pooling_forward::primitive_desc
 
struct  mkldnn::pooling_forward
 
struct  mkldnn::pooling_backward::desc
 
struct  mkldnn::pooling_backward::primitive_desc
 
struct  mkldnn::pooling_backward
 
struct  mkldnn::eltwise_forward::desc
 
struct  mkldnn::eltwise_forward::primitive_desc
 
struct  mkldnn::eltwise_forward
 
struct  mkldnn::eltwise_backward::desc
 
struct  mkldnn::eltwise_backward::primitive_desc
 
struct  mkldnn::eltwise_backward
 
struct  mkldnn::softmax_forward::desc
 
struct  mkldnn::softmax_forward::primitive_desc
 
struct  mkldnn::softmax_forward
 
struct  mkldnn::batch_normalization_forward::desc
 
struct  mkldnn::batch_normalization_forward::primitive_desc
 
struct  mkldnn::batch_normalization_forward
 
struct  mkldnn::batch_normalization_backward::desc
 
struct  mkldnn::batch_normalization_backward::primitive_desc
 
struct  mkldnn::batch_normalization_backward
 
struct  mkldnn::inner_product_forward::desc
 
struct  mkldnn::inner_product_forward::primitive_desc
 
struct  mkldnn::inner_product_forward
 
struct  mkldnn::inner_product_backward_data::desc
 
struct  mkldnn::inner_product_backward_data::primitive_desc
 
struct  mkldnn::inner_product_backward_data
 
struct  mkldnn::inner_product_backward_weights::desc
 
struct  mkldnn::inner_product_backward_weights::primitive_desc
 
struct  mkldnn::inner_product_backward_weights
 

Typedefs

typedef std::vector< std::remove_extent< mkldnn_dims_t >::type > mkldnn::memory::dims
 
typedef eltwise_forward mkldnn::relu_forward
 
typedef eltwise_backward mkldnn::relu_backward
 

Enumerations

enum  mkldnn::engine::kind { mkldnn::engine::any = mkldnn_any_engine, mkldnn::engine::cpu = mkldnn_cpu }
 Kinds of engines. More...
 
enum  mkldnn::memory::data_type {
  mkldnn::memory::data_undef = mkldnn_data_type_undef, mkldnn::memory::f32 = mkldnn_f32, mkldnn::memory::s32 = mkldnn_s32, mkldnn::memory::s16 = mkldnn_s16,
  mkldnn::memory::s8 = mkldnn_s8, mkldnn::memory::u8 = mkldnn_u8
}
 Data type specification. See mkldnn_data_type_t for a detailed description. More...
 
enum  mkldnn::memory::format {
  mkldnn::memory::format_undef = mkldnn_format_undef, mkldnn::memory::any = mkldnn_any, mkldnn::memory::blocked = mkldnn_blocked, mkldnn::memory::x = mkldnn_x,
  mkldnn::memory::nc = mkldnn_nc, mkldnn::memory::nchw = mkldnn_nchw, mkldnn::memory::nhwc = mkldnn_nhwc, mkldnn::memory::chwn = mkldnn_chwn,
  mkldnn::memory::nChw8c = mkldnn_nChw8c, mkldnn::memory::nChw16c = mkldnn_nChw16c, mkldnn::memory::oi = mkldnn_oi, mkldnn::memory::io = mkldnn_io,
  mkldnn::memory::oihw = mkldnn_oihw, mkldnn::memory::ihwo = mkldnn_ihwo, mkldnn::memory::hwio = mkldnn_hwio, mkldnn::memory::oIhw8i = mkldnn_oIhw8i,
  mkldnn::memory::oIhw16i = mkldnn_oIhw16i, mkldnn::memory::OIhw8i8o = mkldnn_OIhw8i8o, mkldnn::memory::OIhw16i16o = mkldnn_OIhw16i16o, mkldnn::memory::OIhw8o8i = mkldnn_OIhw8o8i,
  mkldnn::memory::OIhw16o16i = mkldnn_OIhw16o16i, mkldnn::memory::OIhw8i16o2i = mkldnn_OIhw8i16o2i, mkldnn::memory::OIhw8o16i2o = mkldnn_OIhw8o16i2o, mkldnn::memory::Oihw8o = mkldnn_Oihw8o,
  mkldnn::memory::Oihw16o = mkldnn_Oihw16o, mkldnn::memory::Ohwi8o = mkldnn_Ohwi8o, mkldnn::memory::Ohwi16o = mkldnn_Ohwi16o, mkldnn::memory::OhIw16o4i = mkldnn_OhIw16o4i,
  mkldnn::memory::goihw = mkldnn_goihw, mkldnn::memory::gOIhw8i8o = mkldnn_gOIhw8i8o, mkldnn::memory::gOIhw16i16o = mkldnn_gOIhw16i16o, mkldnn::memory::gOIhw8i16o2i = mkldnn_gOIhw8i16o2i,
  mkldnn::memory::gOIhw8o16i2o = mkldnn_gOIhw8o16i2o, mkldnn::memory::gOihw8o = mkldnn_gOihw8o, mkldnn::memory::gOihw16o = mkldnn_gOihw16o, mkldnn::memory::gOhwi8o = mkldnn_gOhwi8o,
  mkldnn::memory::gOhwi16o = mkldnn_gOhwi16o, mkldnn::memory::gOIhw8o8i = mkldnn_gOIhw8o8i, mkldnn::memory::gOIhw16o16i = mkldnn_gOIhw16o16i, mkldnn::memory::gOhIw16o4i = mkldnn_gOhIw16o4i
}
 Memory format specification. See mkldnn_memory_format_t for a detailed description. More...
 
enum  mkldnn::padding_kind { mkldnn::zero = mkldnn_padding_zero }
 
enum  mkldnn::prop_kind {
  mkldnn::forward_training = mkldnn_forward_training, mkldnn::forward_scoring = mkldnn_forward_scoring, mkldnn::forward_inference = mkldnn_forward_inference, mkldnn::forward = mkldnn_forward,
  mkldnn::backward = mkldnn_backward, mkldnn::backward_data = mkldnn_backward_data, mkldnn::backward_weights = mkldnn_backward_weights, mkldnn::backward_bias = mkldnn_backward_bias
}
 
enum  mkldnn::algorithm {
  mkldnn::convolution_direct = mkldnn_convolution_direct, mkldnn::convolution_winograd = mkldnn_convolution_winograd, mkldnn::eltwise_relu = mkldnn_eltwise_relu, mkldnn::eltwise_tanh = mkldnn_eltwise_tanh,
  mkldnn::eltwise_elu = mkldnn_eltwise_elu, mkldnn::lrn_across_channels = mkldnn_lrn_across_channels, mkldnn::lrn_within_channel = mkldnn_lrn_within_channel, mkldnn::pooling_max = mkldnn_pooling_max,
  mkldnn::pooling_avg = mkldnn_pooling_avg, mkldnn::pooling_avg_include_padding = mkldnn_pooling_avg_include_padding, mkldnn::pooling_avg_exclude_padding = mkldnn_pooling_avg_exclude_padding
}
 
enum  mkldnn::batch_normalization_flag { mkldnn::use_global_stats = mkldnn_use_global_stats, mkldnn::use_scale_shift = mkldnn_use_scaleshift, mkldnn::omit_stats = mkldnn_omit_stats }
 
enum  mkldnn::stream::kind { mkldnn::stream::any = mkldnn_stream_kind_t::mkldnn_any_stream, mkldnn::stream::eager = mkldnn_stream_kind_t::mkldnn_eager, mkldnn::stream::lazy = mkldnn_stream_kind_t::mkldnn_lazy }
 

Functions

 mkldnn::handle< T, traits >::handle (T t=0, bool weak=false)
 Constructs a C handle wrapper. More...
 
bool mkldnn::handle< T, traits >::operator== (const T other) const
 
bool mkldnn::handle< T, traits >::operator!= (const T other) const
 
 mkldnn::handle< T, traits >::handle (const handle &other)
 
handlemkldnn::handle< T, traits >::operator= (const handle &other)
 
void mkldnn::handle< T, traits >::reset (T t, bool weak=false)
 Resets the value of a C handle. More...
 
mkldnn::handle< T, traits >::get () const
 Returns the value of the underlying C handle. More...
 
bool mkldnn::handle< T, traits >::operator== (const handle &other) const
 
bool mkldnn::handle< T, traits >::operator!= (const handle &other) const
 
 mkldnn::primitive::at::at (const primitive &aprimitive, size_t at=0)
 Constructs a wrapper specifying aprimitive output with index at. More...
 
 mkldnn::primitive::at::operator primitive () const
 Returns the specified output. More...
 
const_mkldnn_primitive_desc_t mkldnn::primitive::get_primitive_desc () const
 Returns the descriptor of the underlying C API primitive. More...
 
 mkldnn::error::error (mkldnn_status_t astatus, std::string amessage, mkldnn_primitive_t aerror_primitive=0)
 Constructs an error instance. More...
 
static void mkldnn::error::wrap_c_api (mkldnn_status_t status, std::string message, mkldnn_primitive_t *error_primitive=0)
 A convenience function for wrapping calls to the C API. Checks the return status and throws an error in case of failure. More...
 
static size_t mkldnn::engine::get_count (kind akind)
 Returns the number of engines of a certain kind. More...
 
 mkldnn::engine::engine (kind akind, size_t index)
 Constructs an engine. More...
 
 mkldnn::engine::engine (const mkldnn_engine_t &aengine)
 
 mkldnn::engine::engine (const handle< mkldnn_primitive_desc_t > &pd)
 
template<class primitive_desc >
static engine mkldnn::engine::query (const primitive_desc &pd)
 
template<typename T >
static void mkldnn::memory::validate_dims (std::vector< T > v)
 
 mkldnn::memory::desc::desc (dims adims, data_type adata_type, format aformat)
 Constructs a memory descriptor. More...
 
 mkldnn::memory::desc::desc (const mkldnn_memory_desc_t &adata)
 Constructs a memory descriptor from a C API data structure. More...
 
 mkldnn::memory::primitive_desc::primitive_desc ()
 
 mkldnn::memory::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 Constructs a memory primitive descriptor. More...
 
memory::desc mkldnn::memory::primitive_desc::desc ()
 Returns the memory primitive descriptor. More...
 
size_t mkldnn::memory::primitive_desc::get_size () const
 Returns the number of data elements in the memory described. More...
 
bool mkldnn::memory::primitive_desc::operator== (const primitive_desc &other) const
 
bool mkldnn::memory::primitive_desc::operator!= (const primitive_desc &other) const
 
engine mkldnn::memory::primitive_desc::get_engine ()
 
 mkldnn::memory::memory (const primitive &aprimitive)
 Constructs a memory primitive from a generic primitive. More...
 
 mkldnn::memory::memory (const primitive_desc &adesc)
 Constructs a memory primitive. More...
 
 mkldnn::memory::memory (const primitive_desc &adesc, void *ahandle)
 
primitive_desc mkldnn::memory::get_primitive_desc () const
 Returns the descriptor of the memory primitive. More...
 
void * mkldnn::memory::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 mkldnn::memory::set_data_handle (void *handle) const
 
static mkldnn_data_type_t mkldnn::memory::convert_to_c (data_type adata_type)
 
static mkldnn_memory_format_t mkldnn::memory::convert_to_c (format aformat)
 
bool mkldnn::operator== (mkldnn_data_type_t a, memory::data_type b)
 
bool mkldnn::operator!= (mkldnn_data_type_t a, memory::data_type b)
 
bool mkldnn::operator== (memory::data_type a, mkldnn_data_type_t b)
 
bool mkldnn::operator!= (memory::data_type a, mkldnn_data_type_t b)
 
bool mkldnn::operator== (mkldnn_memory_format_t a, memory::format b)
 
bool mkldnn::operator!= (mkldnn_memory_format_t a, memory::format b)
 
bool mkldnn::operator== (memory::format a, mkldnn_memory_format_t b)
 
bool mkldnn::operator!= (memory::format a, mkldnn_memory_format_t b)
 
mkldnn_padding_kind_t mkldnn::convert_to_c (padding_kind kind)
 
mkldnn_prop_kind_t mkldnn::convert_to_c (prop_kind kind)
 
 mkldnn::reorder::primitive_desc::primitive_desc (const memory::primitive_desc &input, const memory::primitive_desc &output)
 
engine mkldnn::reorder::primitive_desc::get_engine ()
 
 mkldnn::reorder::reorder (const primitive_desc &aprimitive_desc, const primitive::at &input, const memory &output)
 
 mkldnn::reorder::reorder (const primitive::at &input, const memory &output)
 
 mkldnn::view::primitive_desc::primitive_desc (const memory::primitive_desc &input, memory::dims dims, memory::dims offsets)
 
memory::primitive_desc mkldnn::view::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::view::primitive_desc::get_engine ()
 
 mkldnn::view::view (const primitive_desc &view_pd, primitive::at input)
 
 mkldnn::view::view (memory input, memory::dims dims, memory::dims offsets)
 
std::vector< const_mkldnn_primitive_desc_tmkldnn::concat::primitive_desc::cpp_to_c (std::vector< memory::primitive_desc > inputs)
 
 mkldnn::concat::primitive_desc::primitive_desc (const memory::desc &output, int concat_dimension, std::vector< memory::primitive_desc > inputs)
 
 mkldnn::concat::primitive_desc::primitive_desc (int concat_dimension, std::vector< memory::primitive_desc > inputs)
 
memory::primitive_desc mkldnn::concat::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::concat::primitive_desc::get_engine ()
 
 mkldnn::concat::concat (const primitive_desc &concat_pd, std::vector< primitive::at > &inputs, const memory &output)
 
std::vector< const_mkldnn_primitive_desc_tmkldnn::sum::primitive_desc::cpp_to_c (std::vector< memory::primitive_desc > inputs)
 
 mkldnn::sum::primitive_desc::primitive_desc (const memory::desc &output, std::vector< double > scale, std::vector< memory::primitive_desc > inputs)
 
 mkldnn::sum::primitive_desc::primitive_desc (std::vector< double > scale, std::vector< memory::primitive_desc > inputs)
 
memory::primitive_desc mkldnn::sum::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::sum::primitive_desc::get_engine ()
 
 mkldnn::sum::sum (const primitive_desc &sum_pd, std::vector< primitive::at > &inputs, const memory &output)
 
static mkldnn_stream_kind_t mkldnn::stream::convert_to_c (kind akind)
 
 mkldnn::stream::stream (kind akind)
 Constructs a stream. More...
 
streammkldnn::stream::submit (std::vector< primitive > primitives)
 Submits a vector of primitives to a stream for computations. More...
 
bool mkldnn::stream::wait (bool block=true)
 Waits for all computations submitted to the stream to complete. More...
 
streammkldnn::stream::rerun ()
 
 mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims dilates, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims dilates, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::convolution_forward::primitive_desc::src_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_forward::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_forward::primitive_desc::bias_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::convolution_forward::primitive_desc::get_engine ()
 
 mkldnn::convolution_forward::convolution_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const primitive::at &bias, const memory &dst)
 
 mkldnn::convolution_forward::convolution_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst)
 
 mkldnn::convolution_backward_data::desc::desc (algorithm aalgorithm, const memory::desc &diff_src_desc, const memory::desc &weights_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_backward_data::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const convolution_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::diff_src_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::diff_dst_primitive_desc () const
 
engine mkldnn::convolution_backward_data::primitive_desc::get_engine ()
 
 mkldnn::convolution_backward_data::convolution_backward_data (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at &weights, const memory &diff_src)
 
 mkldnn::convolution_backward_weights::desc::desc (algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_backward_weights::desc::desc (algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::convolution_backward_weights::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const convolution_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::src_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_weights_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_bias_primitive_desc () const
 
memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_dst_primitive_desc () const
 
engine mkldnn::convolution_backward_weights::primitive_desc::get_engine ()
 
 mkldnn::convolution_backward_weights::convolution_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_weights, const memory &diff_bias)
 
 mkldnn::convolution_backward_weights::convolution_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_weights)
 
 mkldnn::convolution_relu_forward::desc::desc (const convolution_forward::desc conv_desc, const double negative_slope)
 
 mkldnn::convolution_relu_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
engine mkldnn::convolution_relu_forward::primitive_desc::get_engine ()
 
 mkldnn::convolution_relu_forward::convolution_relu_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const primitive::at &bias, const memory &dst)
 
 mkldnn::convolution_relu_forward::convolution_relu_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst)
 
 mkldnn::lrn_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, int local_size, double alpha, double beta, double k)
 
 mkldnn::lrn_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, int local_size, double alpha, double beta)
 
 mkldnn::lrn_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::lrn_forward::primitive_desc::src_primitive_desc () const
 
memory::primitive_desc mkldnn::lrn_forward::primitive_desc::workspace_primitive_desc () const
 
memory::primitive_desc mkldnn::lrn_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::lrn_forward::primitive_desc::get_engine ()
 
 mkldnn::lrn_forward::lrn_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &workspace, const memory &dst)
 
 mkldnn::lrn_forward::lrn_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst)
 
 mkldnn::lrn_backward::desc::desc (algorithm aalgorithm, const memory::desc &data_desc, const memory::desc &diff_data_desc, int local_size, double alpha, double beta, double k)
 
 mkldnn::lrn_backward::desc::desc (algorithm aalgorithm, const memory::desc &data_desc, const memory::desc &diff_data_desc, int local_size, double alpha, double beta)
 
 mkldnn::lrn_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const lrn_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::lrn_backward::primitive_desc::diff_src_primitive_desc () const
 
memory::primitive_desc mkldnn::lrn_backward::primitive_desc::workspace_primitive_desc () const
 
memory::primitive_desc mkldnn::lrn_backward::primitive_desc::diff_dst_primitive_desc () const
 
engine mkldnn::lrn_backward::primitive_desc::get_engine ()
 
 mkldnn::lrn_backward::lrn_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const primitive::at &workspace, const memory &diff_src)
 
 mkldnn::lrn_backward::lrn_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_src)
 
 mkldnn::pooling_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims kernel, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind)
 
 mkldnn::pooling_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::pooling_forward::primitive_desc::workspace_primitive_desc () const
 
memory::primitive_desc mkldnn::pooling_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::pooling_forward::primitive_desc::get_engine ()
 
 mkldnn::pooling_forward::pooling_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst)
 
 mkldnn::pooling_forward::pooling_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &workspace)
 
 mkldnn::pooling_backward::desc::desc (algorithm aalgorithm, const memory::desc &diff_src_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &kernel, const memory::dims &padding_l, const memory::dims &padding_r, const padding_kind apadding_kind)
 
 mkldnn::pooling_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const pooling_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::pooling_backward::primitive_desc::diff_src_primitive_desc () const
 
engine mkldnn::pooling_backward::primitive_desc::get_engine ()
 
 mkldnn::pooling_backward::pooling_backward (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const memory &diff_src)
 
 mkldnn::pooling_backward::pooling_backward (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at &workspace, const memory &diff_src)
 
template<typename T >
 mkldnn::eltwise_forward::desc::desc (prop_kind aprop_kind, algorithm alg_kind, const memory::desc &src_desc, T alpha=0, T beta=0)
 
template<typename T >
MKLDNN_DEPRECATED mkldnn::eltwise_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, T negative_slope)
 
 mkldnn::eltwise_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::eltwise_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::eltwise_forward::primitive_desc::get_engine ()
 
 mkldnn::eltwise_forward::eltwise_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst)
 
template<typename T >
 mkldnn::eltwise_backward::desc::desc (algorithm alg_kind, const memory::desc &diff_data_desc, const memory::desc &data_desc, T alpha=0, T beta=0)
 
template<typename T >
MKLDNN_DEPRECATED mkldnn::eltwise_backward::desc::desc (const memory::desc &diff_data_desc, const memory::desc &data_desc, T negative_slope)
 
 mkldnn::eltwise_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const eltwise_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::eltwise_backward::primitive_desc::diff_src_primitive_desc () const
 
engine mkldnn::eltwise_backward::primitive_desc::get_engine ()
 
 mkldnn::eltwise_backward::eltwise_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_src)
 
 mkldnn::softmax_forward::desc::desc (prop_kind aprop_kind, const memory::desc &data_desc, int softmax_axis)
 
 mkldnn::softmax_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
engine mkldnn::softmax_forward::primitive_desc::get_engine ()
 
 mkldnn::softmax_forward::softmax_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst)
 
template<typename T >
 mkldnn::batch_normalization_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, T epsilon, unsigned flags)
 
 mkldnn::batch_normalization_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::mean_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::variance_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::batch_normalization_forward::primitive_desc::get_engine ()
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &weights, const memory &dst)
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const memory &dst)
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst, const memory &mean, const memory &variance)
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &mean, const memory &variance)
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst)
 
 mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst)
 
template<typename T >
 mkldnn::batch_normalization_backward::desc::desc (prop_kind aprop_kind, const memory::desc &diff_data_desc, const memory::desc &data_desc, T epsilon, unsigned flags)
 
 mkldnn::batch_normalization_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const batch_normalization_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::diff_weights_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::mean_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::variance_primitive_desc () const
 
memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::batch_normalization_backward::primitive_desc::get_engine ()
 
 mkldnn::batch_normalization_backward::batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const primitive::at &weights, const memory &diff_src, const memory &diff_weights)
 
 mkldnn::batch_normalization_backward::batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const primitive::at &weights, const memory &diff_src)
 
 mkldnn::batch_normalization_backward::batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const memory &diff_src)
 
 mkldnn::inner_product_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc)
 
 mkldnn::inner_product_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc)
 
 mkldnn::inner_product_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine)
 
memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::src_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::bias_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::dst_primitive_desc () const
 
engine mkldnn::inner_product_forward::primitive_desc::get_engine ()
 
 mkldnn::inner_product_forward::inner_product_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at weights, const primitive::at &bias, const memory &dst)
 
 mkldnn::inner_product_forward::inner_product_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at weights, const memory &dst)
 
 mkldnn::inner_product_backward_data::desc::desc (const memory::desc &diff_src_desc, const memory::desc &weights_desc, const memory::desc &diff_dst_desc)
 
 mkldnn::inner_product_backward_data::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const inner_product_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::diff_dst_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::weights_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::diff_src_primitive_desc () const
 
engine mkldnn::inner_product_backward_data::primitive_desc::get_engine ()
 
 mkldnn::inner_product_backward_data::inner_product_backward_data (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at weights, const memory &diff_src)
 
 mkldnn::inner_product_backward_weights::desc::desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc)
 
 mkldnn::inner_product_backward_weights::desc::desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc)
 
 mkldnn::inner_product_backward_weights::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const inner_product_forward::primitive_desc &hint_fwd_primitive_desc)
 
memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_dst_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_weights_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_bias_primitive_desc () const
 
memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::src_primitive_desc () const
 
engine mkldnn::inner_product_backward_weights::primitive_desc::get_engine ()
 
 mkldnn::inner_product_backward_weights::inner_product_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at diff_dst, const memory &diff_weights)
 
 mkldnn::inner_product_backward_weights::inner_product_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at diff_dst, const memory &diff_weights, const memory &diff_bias)
 

Variables

static constexpr auto mkldnn::handle_traits< mkldnn_primitive_t >::destructor = &mkldnn_primitive_destroy
 
mkldnn_primitive_at_t mkldnn::primitive::at::data
 The underlying C API structure. More...
 
mkldnn_status_t mkldnn::error::status
 
std::string mkldnn::error::message
 
primitive mkldnn::error::error_primitive
 
static constexpr auto mkldnn::handle_traits< mkldnn_primitive_desc_t >::destructor = &mkldnn_primitive_desc_destroy
 
mkldnn_memory_desc_t mkldnn::memory::desc::data
 The underlying C API data structure. More...
 
mkldnn_convolution_desc_t mkldnn::convolution_forward::desc::data
 
mkldnn_convolution_desc_t mkldnn::convolution_backward_data::desc::data
 
mkldnn_convolution_desc_t mkldnn::convolution_backward_weights::desc::data
 
mkldnn_convolution_relu_desc_t mkldnn::convolution_relu_forward::desc::data
 
mkldnn_lrn_desc_t mkldnn::lrn_forward::desc::data
 
mkldnn_lrn_desc_t mkldnn::lrn_backward::desc::data
 
mkldnn_pooling_desc_t mkldnn::pooling_forward::desc::data
 
mkldnn_pooling_desc_t mkldnn::pooling_backward::desc::data
 
mkldnn_eltwise_desc_t mkldnn::eltwise_forward::desc::data
 
mkldnn_eltwise_desc_t mkldnn::eltwise_backward::desc::data
 
mkldnn_softmax_desc_t mkldnn::softmax_forward::desc::data
 
mkldnn_batch_normalization_desc_t mkldnn::batch_normalization_forward::desc::data
 
mkldnn_batch_normalization_desc_t mkldnn::batch_normalization_backward::desc::data
 
mkldnn_inner_product_desc_t mkldnn::inner_product_forward::desc::data
 
mkldnn_inner_product_desc_t mkldnn::inner_product_backward_data::desc::data
 
mkldnn_inner_product_desc_t mkldnn::inner_product_backward_weights::desc::data
 

Friends

struct mkldnn::primitive::error
 
struct mkldnn::primitive::stream
 
class mkldnn::primitive::primitive_at
 
class mkldnn::engine::primitive
 
struct mkldnn::memory::desc::memory
 
struct mkldnn::memory::primitive_desc::memory
 

Detailed Description

Typedef Documentation

◆ dims

typedef std::vector<std::remove_extent<mkldnn_dims_t>::type> mkldnn::memory::dims

◆ relu_backward

◆ relu_forward

Enumeration Type Documentation

◆ algorithm

Enumerator
convolution_direct 
convolution_winograd 
eltwise_relu 
eltwise_tanh 
eltwise_elu 
lrn_across_channels 
lrn_within_channel 
pooling_max 
pooling_avg 
pooling_avg_include_padding 
pooling_avg_exclude_padding 

◆ batch_normalization_flag

Enumerator
use_global_stats 
use_scale_shift 
omit_stats 

◆ data_type

Data type specification. See mkldnn_data_type_t for a detailed description.

Enumerator
data_undef 
f32 
s32 
s16 
s8 
u8 

◆ format

Memory format specification. See mkldnn_memory_format_t for a detailed description.

Enumerator
format_undef 
any 
blocked 
nc 
nchw 
nhwc 
chwn 
nChw8c 
nChw16c 
oi 
io 
oihw 
ihwo 
hwio 
oIhw8i 
oIhw16i 
OIhw8i8o 
OIhw16i16o 
OIhw8o8i 
OIhw16o16i 
OIhw8i16o2i 
OIhw8o16i2o 
Oihw8o 
Oihw16o 
Ohwi8o 
Ohwi16o 
OhIw16o4i 
goihw 
gOIhw8i8o 
gOIhw16i16o 
gOIhw8i16o2i 
gOIhw8o16i2o 
gOihw8o 
gOihw16o 
gOhwi8o 
gOhwi16o 
gOIhw8o8i 
gOIhw16o16i 
gOhIw16o4i 

◆ kind [1/2]

Kinds of engines.

Enumerator
any 

An unspecified engine.

cpu 

CPU engine.

◆ kind [2/2]

Enumerator
any 
eager 
lazy 

◆ padding_kind

Enumerator
zero 

◆ prop_kind

Enumerator
forward_training 
forward_scoring 
forward_inference 
forward 
backward 
backward_data 
backward_weights 
backward_bias 

Function Documentation

◆ at()

mkldnn::primitive::at::at ( const primitive aprimitive,
size_t  at = 0 
)
inline

Constructs a wrapper specifying aprimitive output with index at.

Parameters
aprimitiveThe target primitive.
atThe output index.

◆ batch_normalization_backward() [1/3]

mkldnn::batch_normalization_backward::batch_normalization_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at mean,
const primitive::at variance,
const primitive::at diff_dst,
const primitive::at weights,
const memory diff_src,
const memory diff_weights 
)
inline

◆ batch_normalization_backward() [2/3]

mkldnn::batch_normalization_backward::batch_normalization_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at mean,
const primitive::at variance,
const primitive::at diff_dst,
const primitive::at weights,
const memory diff_src 
)
inline

◆ batch_normalization_backward() [3/3]

mkldnn::batch_normalization_backward::batch_normalization_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at mean,
const primitive::at variance,
const primitive::at diff_dst,
const memory diff_src 
)
inline

◆ batch_normalization_forward() [1/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at mean,
const primitive::at variance,
const primitive::at weights,
const memory dst 
)
inline

◆ batch_normalization_forward() [2/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at mean,
const primitive::at variance,
const memory dst 
)
inline

◆ batch_normalization_forward() [3/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const memory dst,
const memory mean,
const memory variance 
)
inline

◆ batch_normalization_forward() [4/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst,
const memory mean,
const memory variance 
)
inline

◆ batch_normalization_forward() [5/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const memory dst 
)
inline

◆ batch_normalization_forward() [6/6]

mkldnn::batch_normalization_forward::batch_normalization_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst 
)
inline

◆ bias_primitive_desc() [1/2]

memory::primitive_desc mkldnn::convolution_forward::primitive_desc::bias_primitive_desc ( ) const
inline

◆ bias_primitive_desc() [2/2]

memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::bias_primitive_desc ( ) const
inline

◆ concat()

mkldnn::concat::concat ( const primitive_desc concat_pd,
std::vector< primitive::at > &  inputs,
const memory output 
)
inline

◆ convert_to_c() [1/5]

static mkldnn_data_type_t mkldnn::memory::convert_to_c ( data_type  adata_type)
inlinestatic

◆ convert_to_c() [2/5]

static mkldnn_memory_format_t mkldnn::memory::convert_to_c ( format  aformat)
inlinestatic

◆ convert_to_c() [3/5]

mkldnn_padding_kind_t mkldnn::convert_to_c ( padding_kind  kind)
inline

◆ convert_to_c() [4/5]

mkldnn_prop_kind_t mkldnn::convert_to_c ( prop_kind  kind)
inline

◆ convert_to_c() [5/5]

static mkldnn_stream_kind_t mkldnn::stream::convert_to_c ( kind  akind)
inlinestatic

◆ convolution_backward_data()

mkldnn::convolution_backward_data::convolution_backward_data ( const primitive_desc aprimitive_desc,
const primitive::at diff_dst,
const primitive::at weights,
const memory diff_src 
)
inline

◆ convolution_backward_weights() [1/2]

mkldnn::convolution_backward_weights::convolution_backward_weights ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at diff_dst,
const memory diff_weights,
const memory diff_bias 
)
inline

◆ convolution_backward_weights() [2/2]

mkldnn::convolution_backward_weights::convolution_backward_weights ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at diff_dst,
const memory diff_weights 
)
inline

◆ convolution_forward() [1/2]

mkldnn::convolution_forward::convolution_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const primitive::at bias,
const memory dst 
)
inline

◆ convolution_forward() [2/2]

mkldnn::convolution_forward::convolution_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const memory dst 
)
inline

◆ convolution_relu_forward() [1/2]

mkldnn::convolution_relu_forward::convolution_relu_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const primitive::at bias,
const memory dst 
)
inline

◆ convolution_relu_forward() [2/2]

mkldnn::convolution_relu_forward::convolution_relu_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at weights,
const memory dst 
)
inline

◆ cpp_to_c() [1/2]

std::vector<const_mkldnn_primitive_desc_t> mkldnn::concat::primitive_desc::cpp_to_c ( std::vector< memory::primitive_desc inputs)
inline

◆ cpp_to_c() [2/2]

std::vector<const_mkldnn_primitive_desc_t> mkldnn::sum::primitive_desc::cpp_to_c ( std::vector< memory::primitive_desc inputs)
inline

◆ desc() [1/29]

mkldnn::memory::desc::desc ( dims  adims,
data_type  adata_type,
format  aformat 
)
inline

Constructs a memory descriptor.

Parameters
adimsData dimensions
adata_typeData precision/type.
aformatData layout format.

◆ desc() [2/29]

mkldnn::memory::desc::desc ( const mkldnn_memory_desc_t adata)
inline

Constructs a memory descriptor from a C API data structure.

Parameters
adataA C API mkldnn_memory_desc_t structure.

◆ desc() [3/29]

memory::desc mkldnn::memory::primitive_desc::desc ( )
inline

Returns the memory primitive descriptor.

◆ desc() [4/29]

mkldnn::convolution_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc bias_desc,
const memory::desc dst_desc,
const memory::dims  strides,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [5/29]

mkldnn::convolution_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc dst_desc,
const memory::dims  strides,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [6/29]

mkldnn::convolution_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc bias_desc,
const memory::desc dst_desc,
const memory::dims  strides,
const memory::dims  dilates,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [7/29]

mkldnn::convolution_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc dst_desc,
const memory::dims  strides,
const memory::dims  dilates,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [8/29]

mkldnn::convolution_backward_data::desc::desc ( algorithm  aalgorithm,
const memory::desc diff_src_desc,
const memory::desc weights_desc,
const memory::desc diff_dst_desc,
const memory::dims  strides,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [9/29]

mkldnn::convolution_backward_weights::desc::desc ( algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc diff_weights_desc,
const memory::desc diff_bias_desc,
const memory::desc diff_dst_desc,
const memory::dims  strides,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [10/29]

mkldnn::convolution_backward_weights::desc::desc ( algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc diff_weights_desc,
const memory::desc diff_dst_desc,
const memory::dims  strides,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [11/29]

mkldnn::convolution_relu_forward::desc::desc ( const convolution_forward::desc  conv_desc,
const double  negative_slope 
)
inline

◆ desc() [12/29]

mkldnn::lrn_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
int  local_size,
double  alpha,
double  beta,
double  k 
)
inline

◆ desc() [13/29]

mkldnn::lrn_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
int  local_size,
double  alpha,
double  beta 
)
inline

◆ desc() [14/29]

mkldnn::lrn_backward::desc::desc ( algorithm  aalgorithm,
const memory::desc data_desc,
const memory::desc diff_data_desc,
int  local_size,
double  alpha,
double  beta,
double  k 
)
inline

◆ desc() [15/29]

mkldnn::lrn_backward::desc::desc ( algorithm  aalgorithm,
const memory::desc data_desc,
const memory::desc diff_data_desc,
int  local_size,
double  alpha,
double  beta 
)
inline

◆ desc() [16/29]

mkldnn::pooling_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  aalgorithm,
const memory::desc src_desc,
const memory::desc dst_desc,
const memory::dims  strides,
const memory::dims  kernel,
const memory::dims  padding_l,
const memory::dims  padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [17/29]

mkldnn::pooling_backward::desc::desc ( algorithm  aalgorithm,
const memory::desc diff_src_desc,
const memory::desc diff_dst_desc,
const memory::dims strides,
const memory::dims kernel,
const memory::dims padding_l,
const memory::dims padding_r,
const padding_kind  apadding_kind 
)
inline

◆ desc() [18/29]

template<typename T >
mkldnn::eltwise_forward::desc::desc ( prop_kind  aprop_kind,
algorithm  alg_kind,
const memory::desc src_desc,
alpha = 0,
beta = 0 
)
inline

◆ desc() [19/29]

template<typename T >
MKLDNN_DEPRECATED mkldnn::eltwise_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc src_desc,
negative_slope 
)
inline
Deprecated:
: api backward compatibility for relu

◆ desc() [20/29]

template<typename T >
mkldnn::eltwise_backward::desc::desc ( algorithm  alg_kind,
const memory::desc diff_data_desc,
const memory::desc data_desc,
alpha = 0,
beta = 0 
)
inline

◆ desc() [21/29]

template<typename T >
MKLDNN_DEPRECATED mkldnn::eltwise_backward::desc::desc ( const memory::desc diff_data_desc,
const memory::desc data_desc,
negative_slope 
)
inline
Deprecated:
: api backward compatibility for relu

◆ desc() [22/29]

mkldnn::softmax_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc data_desc,
int  softmax_axis 
)
inline

◆ desc() [23/29]

template<typename T >
mkldnn::batch_normalization_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc src_desc,
epsilon,
unsigned  flags 
)
inline

◆ desc() [24/29]

template<typename T >
mkldnn::batch_normalization_backward::desc::desc ( prop_kind  aprop_kind,
const memory::desc diff_data_desc,
const memory::desc data_desc,
epsilon,
unsigned  flags 
)
inline

◆ desc() [25/29]

mkldnn::inner_product_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc bias_desc,
const memory::desc dst_desc 
)
inline

◆ desc() [26/29]

mkldnn::inner_product_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc src_desc,
const memory::desc weights_desc,
const memory::desc dst_desc 
)
inline

◆ desc() [27/29]

mkldnn::inner_product_backward_data::desc::desc ( const memory::desc diff_src_desc,
const memory::desc weights_desc,
const memory::desc diff_dst_desc 
)
inline

◆ desc() [28/29]

mkldnn::inner_product_backward_weights::desc::desc ( const memory::desc src_desc,
const memory::desc diff_weights_desc,
const memory::desc diff_bias_desc,
const memory::desc diff_dst_desc 
)
inline

◆ desc() [29/29]

mkldnn::inner_product_backward_weights::desc::desc ( const memory::desc src_desc,
const memory::desc diff_weights_desc,
const memory::desc diff_dst_desc 
)
inline

◆ diff_bias_primitive_desc() [1/2]

memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_bias_primitive_desc ( ) const
inline

◆ diff_bias_primitive_desc() [2/2]

memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_bias_primitive_desc ( ) const
inline

◆ diff_dst_primitive_desc() [1/5]

memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::diff_dst_primitive_desc ( ) const
inline

◆ diff_dst_primitive_desc() [2/5]

memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_dst_primitive_desc ( ) const
inline

◆ diff_dst_primitive_desc() [3/5]

memory::primitive_desc mkldnn::lrn_backward::primitive_desc::diff_dst_primitive_desc ( ) const
inline

◆ diff_dst_primitive_desc() [4/5]

memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::diff_dst_primitive_desc ( ) const
inline

◆ diff_dst_primitive_desc() [5/5]

memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_dst_primitive_desc ( ) const
inline

◆ diff_src_primitive_desc() [1/5]

memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::diff_src_primitive_desc ( ) const
inline

◆ diff_src_primitive_desc() [2/5]

memory::primitive_desc mkldnn::lrn_backward::primitive_desc::diff_src_primitive_desc ( ) const
inline

◆ diff_src_primitive_desc() [3/5]

memory::primitive_desc mkldnn::pooling_backward::primitive_desc::diff_src_primitive_desc ( ) const
inline

◆ diff_src_primitive_desc() [4/5]

memory::primitive_desc mkldnn::eltwise_backward::primitive_desc::diff_src_primitive_desc ( ) const
inline

◆ diff_src_primitive_desc() [5/5]

memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::diff_src_primitive_desc ( ) const
inline

◆ diff_weights_primitive_desc() [1/3]

memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::diff_weights_primitive_desc ( ) const
inline

◆ diff_weights_primitive_desc() [2/3]

memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::diff_weights_primitive_desc ( ) const
inline

◆ diff_weights_primitive_desc() [3/3]

memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::diff_weights_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [1/10]

memory::primitive_desc mkldnn::view::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [2/10]

memory::primitive_desc mkldnn::concat::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [3/10]

memory::primitive_desc mkldnn::sum::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [4/10]

memory::primitive_desc mkldnn::convolution_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [5/10]

memory::primitive_desc mkldnn::lrn_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [6/10]

memory::primitive_desc mkldnn::pooling_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [7/10]

memory::primitive_desc mkldnn::eltwise_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [8/10]

memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [9/10]

memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ dst_primitive_desc() [10/10]

memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::dst_primitive_desc ( ) const
inline

◆ eltwise_backward()

mkldnn::eltwise_backward::eltwise_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at diff_dst,
const memory diff_src 
)
inline

◆ eltwise_forward()

mkldnn::eltwise_forward::eltwise_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst 
)
inline

◆ engine() [1/3]

mkldnn::engine::engine ( kind  akind,
size_t  index 
)
inline

Constructs an engine.

Parameters
akindThe kind of engine to construct.
indexThe index of the engine. Must be less than the value returned by get_count() for this particular kind of engine.

◆ engine() [2/3]

mkldnn::engine::engine ( const mkldnn_engine_t aengine)
inlineexplicit

◆ engine() [3/3]

mkldnn::engine::engine ( const handle< mkldnn_primitive_desc_t > &  pd)
inline

◆ error()

mkldnn::error::error ( mkldnn_status_t  astatus,
std::string  amessage,
mkldnn_primitive_t  aerror_primitive = 0 
)
inline

Constructs an error instance.

Parameters
astatusThe error status returned by the C API.
amessageThe error message.
aerror_primitive(optional) A C handle of the primitive that caused the error.

◆ get()

template<typename T, typename traits = handle_traits<T>>
T mkldnn::handle< T, traits >::get ( ) const
inline

Returns the value of the underlying C handle.

◆ get_count()

static size_t mkldnn::engine::get_count ( kind  akind)
inlinestatic

Returns the number of engines of a certain kind.

Parameters
akindThe kind of engines to count.

◆ get_data_handle()

void* mkldnn::memory::get_data_handle ( ) const
inline

Returns a handle of the data contained in the memory primitive. On the CPU engine, this is a pointer to the allocated memory.

◆ get_engine() [1/21]

engine mkldnn::memory::primitive_desc::get_engine ( )
inline

◆ get_engine() [2/21]

engine mkldnn::reorder::primitive_desc::get_engine ( )
inline

◆ get_engine() [3/21]

engine mkldnn::view::primitive_desc::get_engine ( )
inline

◆ get_engine() [4/21]

engine mkldnn::concat::primitive_desc::get_engine ( )
inline

◆ get_engine() [5/21]

engine mkldnn::sum::primitive_desc::get_engine ( )
inline

◆ get_engine() [6/21]

engine mkldnn::convolution_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [7/21]

engine mkldnn::convolution_backward_data::primitive_desc::get_engine ( )
inline

◆ get_engine() [8/21]

engine mkldnn::convolution_backward_weights::primitive_desc::get_engine ( )
inline

◆ get_engine() [9/21]

engine mkldnn::convolution_relu_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [10/21]

engine mkldnn::lrn_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [11/21]

engine mkldnn::lrn_backward::primitive_desc::get_engine ( )
inline

◆ get_engine() [12/21]

engine mkldnn::pooling_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [13/21]

engine mkldnn::pooling_backward::primitive_desc::get_engine ( )
inline

◆ get_engine() [14/21]

engine mkldnn::eltwise_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [15/21]

engine mkldnn::eltwise_backward::primitive_desc::get_engine ( )
inline

◆ get_engine() [16/21]

engine mkldnn::softmax_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [17/21]

engine mkldnn::batch_normalization_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [18/21]

engine mkldnn::batch_normalization_backward::primitive_desc::get_engine ( )
inline

◆ get_engine() [19/21]

engine mkldnn::inner_product_forward::primitive_desc::get_engine ( )
inline

◆ get_engine() [20/21]

engine mkldnn::inner_product_backward_data::primitive_desc::get_engine ( )
inline

◆ get_engine() [21/21]

engine mkldnn::inner_product_backward_weights::primitive_desc::get_engine ( )
inline

◆ get_primitive_desc() [1/2]

const_mkldnn_primitive_desc_t mkldnn::primitive::get_primitive_desc ( ) const
inline

Returns the descriptor of the underlying C API primitive.

◆ get_primitive_desc() [2/2]

primitive_desc mkldnn::memory::get_primitive_desc ( ) const
inline

Returns the descriptor of the memory primitive.

◆ get_size()

size_t mkldnn::memory::primitive_desc::get_size ( ) const
inline

Returns the number of data elements in the memory described.

Returns the number of bytes required to allocate the memory described including the padding area.

◆ handle() [1/2]

template<typename T, typename traits = handle_traits<T>>
mkldnn::handle< T, traits >::handle ( t = 0,
bool  weak = false 
)
inlineprotected

Constructs a C handle wrapper.

Parameters
tThe C handle to wrap.
weakA flag to specify whether to construct a weak wrapper.

◆ handle() [2/2]

template<typename T, typename traits = handle_traits<T>>
mkldnn::handle< T, traits >::handle ( const handle< T, traits > &  other)
inline

◆ inner_product_backward_data()

mkldnn::inner_product_backward_data::inner_product_backward_data ( const primitive_desc aprimitive_desc,
const primitive::at diff_dst,
const primitive::at  weights,
const memory diff_src 
)
inline

◆ inner_product_backward_weights() [1/2]

mkldnn::inner_product_backward_weights::inner_product_backward_weights ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at  diff_dst,
const memory diff_weights 
)
inline

◆ inner_product_backward_weights() [2/2]

mkldnn::inner_product_backward_weights::inner_product_backward_weights ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at  diff_dst,
const memory diff_weights,
const memory diff_bias 
)
inline

◆ inner_product_forward() [1/2]

mkldnn::inner_product_forward::inner_product_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at  weights,
const primitive::at bias,
const memory dst 
)
inline

◆ inner_product_forward() [2/2]

mkldnn::inner_product_forward::inner_product_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at  weights,
const memory dst 
)
inline

◆ lrn_backward() [1/2]

mkldnn::lrn_backward::lrn_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at diff_dst,
const primitive::at workspace,
const memory diff_src 
)
inline

◆ lrn_backward() [2/2]

mkldnn::lrn_backward::lrn_backward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const primitive::at diff_dst,
const memory diff_src 
)
inline

◆ lrn_forward() [1/2]

mkldnn::lrn_forward::lrn_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory workspace,
const memory dst 
)
inline

◆ lrn_forward() [2/2]

mkldnn::lrn_forward::lrn_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst 
)
inline

◆ mean_primitive_desc() [1/2]

memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::mean_primitive_desc ( ) const
inline

◆ mean_primitive_desc() [2/2]

memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::mean_primitive_desc ( ) const
inline

◆ memory() [1/3]

mkldnn::memory::memory ( const primitive aprimitive)
inline

Constructs a memory primitive from a generic primitive.

Parameters
aprimitiveThe primitive to treat as memory.

◆ memory() [2/3]

mkldnn::memory::memory ( const primitive_desc adesc)
inline

Constructs a memory primitive.

Parameters
adescMemory primitive descriptor.

◆ memory() [3/3]

mkldnn::memory::memory ( const primitive_desc adesc,
void *  ahandle 
)
inline

◆ operator primitive()

mkldnn::primitive::at::operator primitive ( ) const
inline

Returns the specified output.

◆ operator!=() [1/7]

template<typename T, typename traits = handle_traits<T>>
bool mkldnn::handle< T, traits >::operator!= ( const T  other) const
inlineprotected

◆ operator!=() [2/7]

template<typename T, typename traits = handle_traits<T>>
bool mkldnn::handle< T, traits >::operator!= ( const handle< T, traits > &  other) const
inline

◆ operator!=() [3/7]

bool mkldnn::memory::primitive_desc::operator!= ( const primitive_desc other) const
inline

◆ operator!=() [4/7]

bool mkldnn::operator!= ( mkldnn_data_type_t  a,
memory::data_type  b 
)
inline

◆ operator!=() [5/7]

bool mkldnn::operator!= ( memory::data_type  a,
mkldnn_data_type_t  b 
)
inline

◆ operator!=() [6/7]

bool mkldnn::operator!= ( mkldnn_memory_format_t  a,
memory::format  b 
)
inline

◆ operator!=() [7/7]

bool mkldnn::operator!= ( memory::format  a,
mkldnn_memory_format_t  b 
)
inline

◆ operator=()

template<typename T, typename traits = handle_traits<T>>
handle& mkldnn::handle< T, traits >::operator= ( const handle< T, traits > &  other)
inline

◆ operator==() [1/7]

template<typename T, typename traits = handle_traits<T>>
bool mkldnn::handle< T, traits >::operator== ( const T  other) const
inlineprotected

◆ operator==() [2/7]

template<typename T, typename traits = handle_traits<T>>
bool mkldnn::handle< T, traits >::operator== ( const handle< T, traits > &  other) const
inline

◆ operator==() [3/7]

bool mkldnn::memory::primitive_desc::operator== ( const primitive_desc other) const
inline

◆ operator==() [4/7]

bool mkldnn::operator== ( mkldnn_data_type_t  a,
memory::data_type  b 
)
inline

◆ operator==() [5/7]

bool mkldnn::operator== ( memory::data_type  a,
mkldnn_data_type_t  b 
)
inline

◆ operator==() [6/7]

bool mkldnn::operator== ( mkldnn_memory_format_t  a,
memory::format  b 
)
inline

◆ operator==() [7/7]

bool mkldnn::operator== ( memory::format  a,
mkldnn_memory_format_t  b 
)
inline

◆ pooling_backward() [1/2]

mkldnn::pooling_backward::pooling_backward ( const primitive_desc aprimitive_desc,
const primitive::at diff_dst,
const memory diff_src 
)
inline

◆ pooling_backward() [2/2]

mkldnn::pooling_backward::pooling_backward ( const primitive_desc aprimitive_desc,
const primitive::at diff_dst,
const primitive::at workspace,
const memory diff_src 
)
inline

◆ pooling_forward() [1/2]

mkldnn::pooling_forward::pooling_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst 
)
inline

◆ pooling_forward() [2/2]

mkldnn::pooling_forward::pooling_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst,
const memory workspace 
)
inline

◆ primitive_desc() [1/24]

mkldnn::memory::primitive_desc::primitive_desc ( )
inline

◆ primitive_desc() [2/24]

mkldnn::memory::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

Constructs a memory primitive descriptor.

◆ primitive_desc() [3/24]

mkldnn::reorder::primitive_desc::primitive_desc ( const memory::primitive_desc input,
const memory::primitive_desc output 
)
inline

◆ primitive_desc() [4/24]

mkldnn::view::primitive_desc::primitive_desc ( const memory::primitive_desc input,
memory::dims  dims,
memory::dims  offsets 
)
inline

◆ primitive_desc() [5/24]

mkldnn::concat::primitive_desc::primitive_desc ( const memory::desc output,
int  concat_dimension,
std::vector< memory::primitive_desc inputs 
)
inline

◆ primitive_desc() [6/24]

mkldnn::concat::primitive_desc::primitive_desc ( int  concat_dimension,
std::vector< memory::primitive_desc inputs 
)
inline

◆ primitive_desc() [7/24]

mkldnn::sum::primitive_desc::primitive_desc ( const memory::desc output,
std::vector< double >  scale,
std::vector< memory::primitive_desc inputs 
)
inline

◆ primitive_desc() [8/24]

mkldnn::sum::primitive_desc::primitive_desc ( std::vector< double >  scale,
std::vector< memory::primitive_desc inputs 
)
inline

◆ primitive_desc() [9/24]

mkldnn::convolution_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [10/24]

mkldnn::convolution_backward_data::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const convolution_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [11/24]

mkldnn::convolution_backward_weights::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const convolution_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [12/24]

mkldnn::convolution_relu_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [13/24]

mkldnn::lrn_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [14/24]

mkldnn::lrn_backward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const lrn_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [15/24]

mkldnn::pooling_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [16/24]

mkldnn::pooling_backward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const pooling_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [17/24]

mkldnn::eltwise_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [18/24]

mkldnn::eltwise_backward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const eltwise_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [19/24]

mkldnn::softmax_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [20/24]

mkldnn::batch_normalization_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [21/24]

mkldnn::batch_normalization_backward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const batch_normalization_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [22/24]

mkldnn::inner_product_forward::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine 
)
inline

◆ primitive_desc() [23/24]

mkldnn::inner_product_backward_data::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const inner_product_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ primitive_desc() [24/24]

mkldnn::inner_product_backward_weights::primitive_desc::primitive_desc ( const desc adesc,
const engine aengine,
const inner_product_forward::primitive_desc hint_fwd_primitive_desc 
)
inline

◆ query()

template<class primitive_desc >
static engine mkldnn::engine::query ( const primitive_desc &  pd)
inlinestatic

◆ reorder() [1/2]

mkldnn::reorder::reorder ( const primitive_desc aprimitive_desc,
const primitive::at input,
const memory output 
)
inline

◆ reorder() [2/2]

mkldnn::reorder::reorder ( const primitive::at input,
const memory output 
)
inline

◆ rerun()

stream& mkldnn::stream::rerun ( )
inline

◆ reset()

template<typename T, typename traits = handle_traits<T>>
void mkldnn::handle< T, traits >::reset ( t,
bool  weak = false 
)
inline

Resets the value of a C handle.

Parameters
tThe new value of the C handle.
weakA flag to specify whether the wrapper should be weak.

◆ set_data_handle()

void mkldnn::memory::set_data_handle ( void *  handle) const
inline

◆ softmax_forward()

mkldnn::softmax_forward::softmax_forward ( const primitive_desc aprimitive_desc,
const primitive::at src,
const memory dst 
)
inline

◆ src_primitive_desc() [1/5]

memory::primitive_desc mkldnn::convolution_forward::primitive_desc::src_primitive_desc ( ) const
inline

◆ src_primitive_desc() [2/5]

memory::primitive_desc mkldnn::convolution_backward_weights::primitive_desc::src_primitive_desc ( ) const
inline

◆ src_primitive_desc() [3/5]

memory::primitive_desc mkldnn::lrn_forward::primitive_desc::src_primitive_desc ( ) const
inline

◆ src_primitive_desc() [4/5]

memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::src_primitive_desc ( ) const
inline

◆ src_primitive_desc() [5/5]

memory::primitive_desc mkldnn::inner_product_backward_weights::primitive_desc::src_primitive_desc ( ) const
inline

◆ stream()

mkldnn::stream::stream ( kind  akind)
inline

Constructs a stream.

◆ submit()

stream& mkldnn::stream::submit ( std::vector< primitive primitives)
inline

Submits a vector of primitives to a stream for computations.

Parameters
primitivesThe vector of primitives to submit.
Returns
The stream.

◆ sum()

mkldnn::sum::sum ( const primitive_desc sum_pd,
std::vector< primitive::at > &  inputs,
const memory output 
)
inline

◆ validate_dims()

template<typename T >
static void mkldnn::memory::validate_dims ( std::vector< T >  v)
inlinestatic

◆ variance_primitive_desc() [1/2]

memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::variance_primitive_desc ( ) const
inline

◆ variance_primitive_desc() [2/2]

memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::variance_primitive_desc ( ) const
inline

◆ view() [1/2]

mkldnn::view::view ( const primitive_desc view_pd,
primitive::at  input 
)
inline

◆ view() [2/2]

mkldnn::view::view ( memory  input,
memory::dims  dims,
memory::dims  offsets 
)
inline

◆ wait()

bool mkldnn::stream::wait ( bool  block = true)
inline

Waits for all computations submitted to the stream to complete.

Parameters
blockSpecifies whether the operation should wait indefinitely or return immediately.
Returns
true if all computations completed.
false if not all computations completed.

◆ weights_primitive_desc() [1/6]

memory::primitive_desc mkldnn::convolution_forward::primitive_desc::weights_primitive_desc ( ) const
inline

◆ weights_primitive_desc() [2/6]

memory::primitive_desc mkldnn::convolution_backward_data::primitive_desc::weights_primitive_desc ( ) const
inline

◆ weights_primitive_desc() [3/6]

memory::primitive_desc mkldnn::batch_normalization_forward::primitive_desc::weights_primitive_desc ( ) const
inline

◆ weights_primitive_desc() [4/6]

memory::primitive_desc mkldnn::batch_normalization_backward::primitive_desc::weights_primitive_desc ( ) const
inline

◆ weights_primitive_desc() [5/6]

memory::primitive_desc mkldnn::inner_product_forward::primitive_desc::weights_primitive_desc ( ) const
inline

◆ weights_primitive_desc() [6/6]

memory::primitive_desc mkldnn::inner_product_backward_data::primitive_desc::weights_primitive_desc ( ) const
inline

◆ workspace_primitive_desc() [1/3]

memory::primitive_desc mkldnn::lrn_forward::primitive_desc::workspace_primitive_desc ( ) const
inline

◆ workspace_primitive_desc() [2/3]

memory::primitive_desc mkldnn::lrn_backward::primitive_desc::workspace_primitive_desc ( ) const
inline

◆ workspace_primitive_desc() [3/3]

memory::primitive_desc mkldnn::pooling_forward::primitive_desc::workspace_primitive_desc ( ) const
inline

◆ wrap_c_api()

static void mkldnn::error::wrap_c_api ( mkldnn_status_t  status,
std::string  message,
mkldnn_primitive_t error_primitive = 0 
)
inlinestatic

A convenience function for wrapping calls to the C API. Checks the return status and throws an error in case of failure.

Parameters
statusThe error status returned by the C API.
messageThe error message.
error_primitive(optional) A C handle of the primitive that caused the error.

Variable Documentation

◆ data [1/18]

mkldnn_primitive_at_t mkldnn::primitive::at::data

The underlying C API structure.

◆ data [2/18]

mkldnn_memory_desc_t mkldnn::memory::desc::data

The underlying C API data structure.

◆ data [3/18]

mkldnn_convolution_desc_t mkldnn::convolution_forward::desc::data

◆ data [4/18]

mkldnn_convolution_desc_t mkldnn::convolution_backward_data::desc::data

◆ data [5/18]

mkldnn_convolution_desc_t mkldnn::convolution_backward_weights::desc::data

◆ data [6/18]

mkldnn_convolution_relu_desc_t mkldnn::convolution_relu_forward::desc::data

◆ data [7/18]

mkldnn_lrn_desc_t mkldnn::lrn_forward::desc::data

◆ data [8/18]

mkldnn_lrn_desc_t mkldnn::lrn_backward::desc::data

◆ data [9/18]

mkldnn_pooling_desc_t mkldnn::pooling_forward::desc::data

◆ data [10/18]

mkldnn_pooling_desc_t mkldnn::pooling_backward::desc::data

◆ data [11/18]

mkldnn_eltwise_desc_t mkldnn::eltwise_forward::desc::data

◆ data [12/18]

mkldnn_eltwise_desc_t mkldnn::eltwise_backward::desc::data

◆ data [13/18]

mkldnn_softmax_desc_t mkldnn::softmax_forward::desc::data

◆ data [14/18]

mkldnn_batch_normalization_desc_t mkldnn::batch_normalization_forward::desc::data

◆ data [15/18]

mkldnn_batch_normalization_desc_t mkldnn::batch_normalization_backward::desc::data

◆ data [16/18]

mkldnn_inner_product_desc_t mkldnn::inner_product_forward::desc::data

◆ data [17/18]

mkldnn_inner_product_desc_t mkldnn::inner_product_backward_data::desc::data

◆ data [18/18]

mkldnn_inner_product_desc_t mkldnn::inner_product_backward_weights::desc::data

◆ destructor [1/2]

constexpr auto mkldnn::handle_traits< mkldnn_primitive_t >::destructor = &mkldnn_primitive_destroy
static

◆ destructor [2/2]

◆ error_primitive

primitive mkldnn::error::error_primitive

◆ message

std::string mkldnn::error::message

◆ status

mkldnn_status_t mkldnn::error::status

Friends

◆ error

friend struct error
friend

◆ memory [1/2]

friend struct memory
friend

◆ memory [2/2]

friend struct memory
friend

◆ primitive

friend class primitive
friend

◆ primitive_at

friend class primitive_at
friend

◆ stream

friend struct stream
friend