|
mxnet
|
|
Functions | |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_create (mkldnn_stream_t *stream, mkldnn_stream_kind_t 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) |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_wait (mkldnn_stream_t stream, int block, mkldnn_primitive_t *error_primitive) |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_rerun (mkldnn_stream_t stream, mkldnn_primitive_t *error_primitive) |
| mkldnn_status_t MKLDNN_API | mkldnn_stream_destroy (mkldnn_stream_t stream) |
| 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_destroy | ( | mkldnn_stream_t | stream | ) |
Destroys an execution stream.
| 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_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.
1.8.11