Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.21.0
Performance library for Deep Learning
|
An execution engine. More...
#include <mkldnn.hpp>
Public Types | |
enum | kind { any = mkldnn_any_engine, cpu = mkldnn_cpu } |
Kinds of engines. More... | |
Public Member Functions | |
engine (kind akind, size_t index) | |
Constructs an engine. More... | |
engine (const mkldnn_engine_t &aengine) | |
engine (const handle< mkldnn_primitive_desc_t > &pd) | |
Public Member Functions inherited from mkldnn::handle< mkldnn_engine_t > | |
handle (mkldnn_engine_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_engine_t t, bool weak=false) |
Resets the value of a C handle. More... | |
mkldnn_engine_t | get () const |
Returns the value of the underlying C handle. More... | |
Static Public Member Functions | |
static size_t | get_count (kind akind) |
Returns the number of engines of a certain kind. More... | |
template<class primitive_desc > | |
static engine | query (const primitive_desc &pd) |
Friends | |
class | primitive |
Additional Inherited Members | |
Protected Member Functions inherited from mkldnn::handle< mkldnn_engine_t > | |
bool | operator== (const mkldnn_engine_t other) const |
bool | operator!= (const mkldnn_engine_t other) const |
An execution engine.
enum mkldnn::engine::kind |
|
inline |
Constructs an engine.
akind | The kind of engine to construct. |
index | The index of the engine. Must be less than the value returned by get_count() for this particular kind of engine. |
|
inlineexplicit |
|
inline |
|
inlinestatic |
Returns the number of engines of a certain kind.
akind | The kind of engines to count. |
|
inlinestatic |
|
friend |