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
.