Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  1.0.4
Performance library for Deep Learning
Public Types | Public Member Functions | List of all members
mkldnn::primitive Class Reference

Base class for all computational primitives. More...

#include <mkldnn.hpp>

Inheritance diagram for mkldnn::primitive:
Inheritance graph
[legend]
Collaboration diagram for mkldnn::primitive:
Collaboration graph
[legend]

Public Types

enum  kind {
  kind::undef = mkldnn_undefined_primitive, kind::reorder = mkldnn_reorder, kind::shuffle = mkldnn_shuffle, kind::concat = mkldnn_concat,
  kind::sum = mkldnn_sum, kind::convolution = mkldnn_convolution, kind::deconvolution = mkldnn_deconvolution, kind::eltwise = mkldnn_eltwise,
  kind::softmax = mkldnn_softmax, kind::pooling = mkldnn_pooling, kind::lrn = mkldnn_lrn, kind::batch_normalization = mkldnn_batch_normalization,
  kind::inner_product = mkldnn_inner_product, kind::rnn = mkldnn_rnn
}
 Kinds of primitives. More...
 

Public Member Functions

const_mkldnn_primitive_desc_t get_primitive_desc () const
 Returns the descriptor of the underlying C API primitive.
 
- Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_t >
 handle (mkldnn_primitive_t t, bool weak=false)
 Constructs a C handle wrapper. More...
 
 handle ()
 Empty constructor. More...
 
void reset (mkldnn_primitive_t t, bool weak=false)
 Resets the value of a C handle. More...
 
mkldnn_primitive_t get (bool allow_emtpy=false) const
 Returns the value of the underlying C handle.
 

Detailed Description

Base class for all computational primitives.

Member Enumeration Documentation

◆ kind

Kinds of primitives.

Used to implement a way to extend the library with new primitives without changing the ABI.

Enumerator
undef 

Undefined primitive.

reorder 

A reorder primitive.

shuffle 

A shuffle primitive.

concat 

A (out-of-place) concat primitive.

sum 

A sum primitive.

convolution 

A convolution primitive.

deconvolution 

A deconvolution primitive.

eltwise 

An element-wise primitive.

softmax 

A softmax primitive.

pooling 

A pooling primitive.

lrn 

An LRN primitive.

batch_normalization 

An batch normalization primitive.

inner_product 

An inner product primitive.

rnn 

A rnn primitive.


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