Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.21.0
Performance library for Deep Learning
|
#include <mkldnn.hpp>
Public Types | |
enum | kind { any = mkldnn_stream_kind_t::mkldnn_any_stream, eager = mkldnn_stream_kind_t::mkldnn_eager, lazy = mkldnn_stream_kind_t::mkldnn_lazy } |
Public Member Functions | |
stream (kind akind) | |
Constructs a stream. More... | |
stream & | submit (std::vector< primitive > primitives) |
Submits a vector of primitives to a stream for computations. More... | |
bool | wait (bool block=true) |
Waits for all computations submitted to the stream to complete. More... | |
stream & | rerun () |
handle (T t=0, bool weak=false) | |
Constructs a C handle wrapper. More... | |
handle (const handle &other) | |
Public Member Functions inherited from mkldnn::handle< mkldnn_stream_t > | |
handle (mkldnn_stream_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_stream_t t, bool weak=false) |
Resets the value of a C handle. More... | |
mkldnn_stream_t | get () const |
Returns the value of the underlying C handle. More... | |
Static Public Member Functions | |
static mkldnn_stream_kind_t | convert_to_c (kind akind) |
Additional Inherited Members | |
Protected Member Functions inherited from mkldnn::handle< mkldnn_stream_t > | |
bool | operator== (const mkldnn_stream_t other) const |
bool | operator!= (const mkldnn_stream_t other) const |
enum mkldnn::stream::kind |
|
inline |
Constructs a stream.
|
inlinestatic |
Submits a vector of primitives to a stream for computations.
primitives | The vector of primitives to submit. |
|
inline |
Waits for all computations submitted to the stream to complete.
block | Specifies whether the operation should wait indefinitely or return immediately. |
true
if all computations completed. false
if not all computations completed.
|
inline |
|
inline |
Constructs a C handle wrapper.
t | The C handle to wrap. |
weak | A flag to specify whether to construct a weak wrapper. |
|
inline |