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
.