|
Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.17
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 () |
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 |
1.8.13