Deep Neural Network Library (DNNL)  1.1.3
Performance library for Deep Learning
Public Types | Public Member Functions | List of all members
dnnl::primitive Class Reference

Base class for all computational primitives. More...

#include <dnnl.hpp>

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

Public Types

enum  kind {
  kind::undef = dnnl_undefined_primitive, kind::reorder = dnnl_reorder, kind::shuffle = dnnl_shuffle, kind::concat = dnnl_concat,
  kind::sum = dnnl_sum, kind::convolution = dnnl_convolution, kind::deconvolution = dnnl_deconvolution, kind::eltwise = dnnl_eltwise,
  kind::softmax = dnnl_softmax, kind::pooling = dnnl_pooling, kind::lrn = dnnl_lrn, kind::batch_normalization = dnnl_batch_normalization,
  kind::layer_normalization = dnnl_layer_normalization, kind::inner_product = dnnl_inner_product, kind::rnn = dnnl_rnn, kind::binary = dnnl_binary
}
 Kinds of primitives. More...
 

Public Member Functions

const_dnnl_primitive_desc_t get_primitive_desc () const
 Returns the descriptor of the underlying C API primitive.
 
- Public Member Functions inherited from dnnl::handle< dnnl_primitive_t >
 handle ()=default
 Empty constructor. More...
 
 handle (dnnl_primitive_t t, bool weak=false)
 Constructs a C handle wrapper from a C handle. More...
 
void reset (dnnl_primitive_t t, bool weak=false)
 Resets the value of a C handle. More...
 
dnnl_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

enum dnnl::primitive::kind
strong

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 

A batch normalization primitive.

layer_normalization 

A layer normalization primitive.

inner_product 

An inner product primitive.

rnn 

A rnn primitive.

binary 

A binary primitive.


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