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