mxnet.engine¶

Engine properties management.

Functions

`bulk`(size)	Bulk execution bundles many operators to run together.
`set_bulk_size`(size)	Set size limit on bulk execution.

mxnet.engine.bulk(size)[source]¶

Bulk execution bundles many operators to run together. This can improve performance when running a lot of small operators sequentially.

Returns a scope for managing bulk size:

with mx.engine.bulk(10):
    x = mx.nd.zeros((1,))
    for i in range(100):
        x += 1

mxnet.engine.set_bulk_size(size)[source]¶

Set size limit on bulk execution.

Bulk execution bundles many operators to run together. This can improve performance when running a lot of small operators sequentially.

Parameters: size (int) – Maximum number of operators that can be bundled in a bulk.
Returns: Previous bulk size.
Return type: int

mxnet / mxnet.engine