|
mxnet
|
|
Classes | |
| struct | dnnl_stream |
Typedefs | |
| typedef struct dnnl_stream * | dnnl_stream_t |
| An execution stream handle. More... | |
| typedef const struct dnnl_stream * | const_dnnl_stream_t |
| A constant execution stream handle. More... | |
| typedef struct dnnl_stream_attr * | dnnl_stream_attr_t |
| An execution stream attributes handle. More... | |
| typedef const struct dnnl_stream_attr * | const_dnnl_stream_attr_t |
| A constant execution stream attributes handle. More... | |
Enumerations | |
| enum | dnnl_stream_flags_t { dnnl_stream_default_order = 0x1U, dnnl_stream_in_order = 0x2U, dnnl_stream_out_of_order = 0x4U, dnnl_stream_default_flags = dnnl_stream_default_order } |
| Stream flags. More... | |
Functions | |
| dnnl_status_t DNNL_API | dnnl_stream_attr_create (dnnl_stream_attr_t *attr, dnnl_engine_kind_t kind) |
| dnnl_status_t DNNL_API | dnnl_stream_attr_destroy (dnnl_stream_attr_t attr) |
| dnnl_status_t DNNL_API | dnnl_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags) |
| dnnl_status_t DNNL_API | dnnl_stream_create_v2 (dnnl_stream_t *stream, dnnl_engine_t engine, unsigned flags, const_dnnl_stream_attr_t attr) |
| dnnl_status_t DNNL_API | dnnl_stream_wait (dnnl_stream_t stream) |
| dnnl_status_t DNNL_API | dnnl_stream_destroy (dnnl_stream_t stream) |
| typedef const struct dnnl_stream_attr* const_dnnl_stream_attr_t |
A constant execution stream attributes handle.
| typedef const struct dnnl_stream* const_dnnl_stream_t |
A constant execution stream handle.
| typedef struct dnnl_stream_attr* dnnl_stream_attr_t |
An execution stream attributes handle.
| typedef struct dnnl_stream* dnnl_stream_t |
An execution stream handle.
| enum dnnl_stream_flags_t |
| dnnl_status_t DNNL_API dnnl_stream_attr_create | ( | dnnl_stream_attr_t * | attr, |
| dnnl_engine_kind_t | kind | ||
| ) |
Creates execution stream attributes for a stream that runs on an engine of a particular kind.
| attr | Output execution stream attributes. |
| kind | Target engine kind. |
| dnnl_status_t DNNL_API dnnl_stream_attr_destroy | ( | dnnl_stream_attr_t | attr | ) |
Destroys execution stream attributes.
| attr | Execution stream attributes to destroy. |
| dnnl_status_t DNNL_API dnnl_stream_create | ( | dnnl_stream_t * | stream, |
| dnnl_engine_t | engine, | ||
| unsigned | flags | ||
| ) |
Creates an execution stream.
| stream | Output execution stream. |
| engine | Engine to create the execution stream on. |
| flags | Stream behavior flags ( |
| dnnl_status_t DNNL_API dnnl_stream_create_v2 | ( | dnnl_stream_t * | stream, |
| dnnl_engine_t | engine, | ||
| unsigned | flags, | ||
| const_dnnl_stream_attr_t | attr | ||
| ) |
Creates an execution stream.
| stream | Output execution stream. |
| engine | Engine to create the execution stream on. |
| flags | Stream behavior flags ( |
| attr | Stream attributes. |
| dnnl_status_t DNNL_API dnnl_stream_destroy | ( | dnnl_stream_t | stream | ) |
Destroys an execution stream.
| stream | Execution stream to destroy. |
| dnnl_status_t DNNL_API dnnl_stream_wait | ( | dnnl_stream_t | stream | ) |
Waits for all primitives in the execution stream to finish computations.
| stream | Execution stream. |
1.8.11