Classes | |
struct | dnnl::engine |
An execution engine. More... | |
struct | dnnl_engine |
An opaque structure to describe an engine. More... | |
Typedefs | |
typedef struct dnnl_engine * | dnnl_engine_t |
An engine handle. | |
Enumerations | |
enum | dnnl_engine_kind_t |
Kinds of engines. More... | |
Functions | |
size_t DNNL_API | dnnl_engine_get_count (dnnl_engine_kind_t kind) |
Returns the number of engines of a particular kind. More... | |
dnnl_status_t DNNL_API | dnnl_engine_create (dnnl_engine_t *engine, dnnl_engine_kind_t kind, size_t index) |
Creates an engine. More... | |
dnnl_status_t DNNL_API | dnnl_engine_get_kind (dnnl_engine_t engine, dnnl_engine_kind_t *kind) |
Returns the kind of an engine. More... | |
dnnl_status_t DNNL_API | dnnl_engine_destroy (dnnl_engine_t engine) |
Destroys an engine. More... | |
dnnl_engine_kind_t | dnnl::convert_to_c (engine::kind akind) |
Converts engine kind enum value from C++ API to C API type. More... | |
An abstraction of a computational device: a CPU, a specific GPU card in the system, etc. Most primitives are created to execute computations on one specific engine. The only exceptions are reorder primitives that transfer data between two different engines.
enum dnnl_engine_kind_t |
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_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_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. |
dnnl_status_t DNNL_API dnnl_engine_destroy | ( | dnnl_engine_t | engine | ) |
Destroys an engine.
engine | Engine to destroy. |
|
inline |
Converts engine kind enum value from C++ API to C API type.
akind | C++ API engine kind enum value. |