|
mxnet
|
|
Functions | |
| size_t MKLDNN_API | mkldnn_engine_get_count (mkldnn_engine_kind_t kind) |
Returns the number of engines of a particular kind. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_create (mkldnn_engine_t *engine, mkldnn_engine_kind_t kind, size_t index) |
Creates an engine of particular kind and index. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_create_ocl (mkldnn_engine_t *engine, mkldnn_engine_kind_t kind, cl_device_id device, cl_context context) |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_get_kind (mkldnn_engine_t engine, mkldnn_engine_kind_t *kind) |
Returns the kind of an engine. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_get_ocl_context (mkldnn_engine_t engine, cl_context *context) |
Returns an OpenCL context associated with an engine. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_get_ocl_device (mkldnn_engine_t engine, cl_device_id *device) |
Returns an OpenCL device associated with an engine. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_engine_destroy (mkldnn_engine_t engine) |
Destroys an engine. More... | |
| mkldnn_status_t MKLDNN_API mkldnn_engine_create | ( | mkldnn_engine_t * | engine, |
| mkldnn_engine_kind_t | kind, | ||
| size_t | index | ||
| ) |
Creates an engine of particular kind and index.
| mkldnn_status_t MKLDNN_API mkldnn_engine_create_ocl | ( | mkldnn_engine_t * | engine, |
| mkldnn_engine_kind_t | kind, | ||
| cl_device_id | device, | ||
| cl_context | context | ||
| ) |
Creates an engine of particular kind associated with a given OpenCL device and context objects.
| mkldnn_status_t MKLDNN_API mkldnn_engine_destroy | ( | mkldnn_engine_t | engine | ) |
Destroys an engine.
| size_t MKLDNN_API mkldnn_engine_get_count | ( | mkldnn_engine_kind_t | kind | ) |
Returns the number of engines of a particular kind.
| mkldnn_status_t MKLDNN_API mkldnn_engine_get_kind | ( | mkldnn_engine_t | engine, |
| mkldnn_engine_kind_t * | kind | ||
| ) |
Returns the kind of an engine.
| mkldnn_status_t MKLDNN_API mkldnn_engine_get_ocl_context | ( | mkldnn_engine_t | engine, |
| cl_context * | context | ||
| ) |
Returns an OpenCL context associated with an engine.
| mkldnn_status_t MKLDNN_API mkldnn_engine_get_ocl_device | ( | mkldnn_engine_t | engine, |
| cl_device_id * | device | ||
| ) |
Returns an OpenCL device associated with an engine.
1.8.11