|
Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.17
Performance library for Deep Learning
|
Functions | |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_create (mkldnn_stream_t *stream, mkldnn_stream_kind_t stream_kind) |
Creates an execution stream of stream_kind. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_submit (mkldnn_stream_t stream, size_t n, mkldnn_primitive_t primitives[], mkldnn_primitive_t *error_primitive) |
Submits primitives to an execution stream. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_wait (mkldnn_stream_t stream, int block, mkldnn_primitive_t *error_primitive) |
Waits for all primitives in the execution stream to finish. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_rerun (mkldnn_stream_t stream, mkldnn_primitive_t *error_primitive) |
Reruns all the primitives within the stream. 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_stream_kind_t | stream_kind | ||
| ) |
Creates an execution stream of stream_kind.
| mkldnn_status_t MKLDNN_API mkldnn_stream_submit | ( | mkldnn_stream_t | stream, |
| size_t | n, | ||
| mkldnn_primitive_t | primitives[], | ||
| mkldnn_primitive_t * | error_primitive | ||
| ) |
Submits primitives to an execution stream.
The number of primitives is n. All or none of the primitives can be lazy. In case of an error, returns the offending error_primitive if it is not NULL.
| mkldnn_status_t MKLDNN_API mkldnn_stream_wait | ( | mkldnn_stream_t | stream, |
| int | block, | ||
| mkldnn_primitive_t * | error_primitive | ||
| ) |
Waits for all primitives in the execution stream to finish.
Returns immediately if block is zero. In case of an error, returns the offending error_primitive if it is not NULL.
| mkldnn_status_t MKLDNN_API mkldnn_stream_rerun | ( | mkldnn_stream_t | stream, |
| mkldnn_primitive_t * | error_primitive | ||
| ) |
Reruns all the primitives within the stream.
In case of an error, returns the offending error_primitive if it is not NULL.
| mkldnn_status_t MKLDNN_API mkldnn_stream_destroy | ( | mkldnn_stream_t | stream | ) |
Destroys an execution stream.
1.8.13