mxnet
|
Functions | |
mkldnn_status_t MKLDNN_API | mkldnn_stream_create (mkldnn_stream_t *stream, mkldnn_engine_t engine, unsigned flags) |
Creates an execution stream for engine and with flags . More... | |
mkldnn_status_t MKLDNN_API | mkldnn_stream_create_ocl (mkldnn_stream_t *stream, mkldnn_engine_t engine, cl_command_queue queue) |
mkldnn_status_t MKLDNN_API | mkldnn_stream_get_ocl_command_queue (mkldnn_stream_t stream, cl_command_queue *queue) |
mkldnn_status_t MKLDNN_API | mkldnn_stream_wait (mkldnn_stream_t stream) |
Waits for all primitives in the execution stream to finish. More... | |
mkldnn_status_t MKLDNN_API | mkldnn_stream_destroy (mkldnn_stream_t stream) |
Destroys an execution stream . More... | |
mkldnn_status_t MKLDNN_API mkldnn_stream_create | ( | mkldnn_stream_t * | stream, |
mkldnn_engine_t | engine, | ||
unsigned | flags | ||
) |
Creates an execution stream
for engine
and with flags
.
mkldnn_status_t MKLDNN_API mkldnn_stream_create_ocl | ( | mkldnn_stream_t * | stream, |
mkldnn_engine_t | engine, | ||
cl_command_queue | queue | ||
) |
Creates an execution stream
for a given engine
associated with an OpenCL command queue
.
mkldnn_status_t MKLDNN_API mkldnn_stream_destroy | ( | mkldnn_stream_t | stream | ) |
Destroys an execution stream
.
mkldnn_status_t MKLDNN_API mkldnn_stream_get_ocl_command_queue | ( | mkldnn_stream_t | stream, |
cl_command_queue * | queue | ||
) |
Returns the OpenCL command queue
associated with an execution stream
.
mkldnn_status_t MKLDNN_API mkldnn_stream_wait | ( | mkldnn_stream_t | stream | ) |
Waits for all primitives in the execution stream
to finish.