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. |