Base class for all computational primitives. More...
#include <mkldnn.hpp>
Inheritance diagram for mkldnn::primitive:
Collaboration diagram for mkldnn::primitive:
Classes | |
| struct | at |
| A wrapper structure to specify a particular output of a primitive. More... | |
Public Types | |
| enum | kind { kind::undefined_primitive = mkldnn_undefined_primitive, kind::memory = mkldnn_memory, kind::view = mkldnn_view, kind::reorder = mkldnn_reorder, kind::concat = mkldnn_concat, kind::concat_inplace = mkldnn_concat_inplace, kind::sum = mkldnn_sum, kind::convolution = mkldnn_convolution, kind::deconvolution = mkldnn_deconvolution, kind::shuffle = mkldnn_shuffle, 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 } |
| A proxy to C primitive kind enum. More... | |
Public Member Functions | |
| const_mkldnn_primitive_desc_t | get_primitive_desc () const |
| Returns the descriptor of the underlying C API primitive. More... | |
 Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_t > | |
| handle (mkldnn_primitive_t t=0, bool weak=false) | |
| Constructs a C handle wrapper. 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... | |
Friends | |
| struct | error |
| struct | stream |
| class | primitive_at |
Additional Inherited Members | |
| Protected Member Functions inherited from mkldnn::handle< mkldnn_primitive_t > | |
| bool | operator== (const mkldnn_primitive_t other) const |
| bool | operator!= (const mkldnn_primitive_t other) const |
Detailed Description
Base class for all computational primitives.
Member Enumeration Documentation
◆ kind
|
strong |
Friends And Related Function Documentation
◆ error
|
friend |
◆ stream
|
friend |
◆ primitive_at
|
friend |
The documentation for this class was generated from the following file:
- include/mkldnn.hpp
