|
mxnet
|
|
Classes | |
| struct | dnnl_engine |
| An opaque structure to describe an engine. More... | |
Typedefs | |
| typedef struct dnnl_engine * | dnnl_engine_t |
| An engine handle. More... | |
Enumerations | |
| enum | dnnl_engine_kind_t { dnnl_any_engine, dnnl_cpu, dnnl_gpu } |
| Kinds of engines. More... | |
Functions | |
| size_t DNNL_API | dnnl_engine_get_count (dnnl_engine_kind_t kind) |
| dnnl_status_t DNNL_API | dnnl_engine_create (dnnl_engine_t *engine, dnnl_engine_kind_t kind, size_t index) |
| dnnl_status_t DNNL_API | dnnl_engine_get_kind (dnnl_engine_t engine, dnnl_engine_kind_t *kind) |
| dnnl_status_t DNNL_API | dnnl_engine_destroy (dnnl_engine_t engine) |
| typedef struct dnnl_engine* dnnl_engine_t |
An engine handle.
| enum dnnl_engine_kind_t |
| dnnl_status_t DNNL_API dnnl_engine_create | ( | dnnl_engine_t * | engine, |
| dnnl_engine_kind_t | kind, | ||
| size_t | index | ||
| ) |
Creates an engine.
| engine | Output engine. |
| kind | Engine kind. |
| index | Engine index that should be between 0 and the count of engines of the requested kind. |
| dnnl_status_t DNNL_API dnnl_engine_destroy | ( | dnnl_engine_t | engine | ) |
Destroys an engine.
| engine | Engine to destroy. |
| size_t DNNL_API dnnl_engine_get_count | ( | dnnl_engine_kind_t | kind | ) |
Returns the number of engines of a particular kind.
| kind | Kind of engines to count. |
| dnnl_status_t DNNL_API dnnl_engine_get_kind | ( | dnnl_engine_t | engine, |
| dnnl_engine_kind_t * | kind | ||
| ) |
Returns the kind of an engine.
| engine | Engine to query. |
| kind | Output engine kind. |
1.8.13