ONNX

Introduction:

This page tracks the current status and development  to support ONNX  on Mxnet. We currently support:

• import of ONNX models into Mxnet.
• export MXNet models to ONNX format.

ONNX-MXNET Import Module

We have two repositories for the import module:

1. onnx-mxnet repository inside onnx https://github.com/onnx/onnx-mxnet  (deprecated)
2. A newly created and refactored module inside Mxnet -> contrib. https://github.com/apache/incubator-mxnet/tree/master/python/mxnet/contrib/onnx/onnx2mx

MXNET-ONNX Export module

A new module has been added to MXNet repository to export MXNet models to ONNX model format. https://github.com/apache/incubator-mxnet/tree/master/python/mxnet/contrib/onnx/mx2onnx

Operator Coverage:

This table keeps track of the status of all ONNX operators supported by Mxnet.
ONNX backend test script reports the coverage on the operators and attributes.

Status meaning:

OK = currently we support the operator

OK, in review = PR is out, waiting for tomorrow’s meeting and feedback

NOT OK, under development = Operator is missing on MXNet backend or direct 1:1 mapping doesn’t exist

NOT OK = Not supported right now

ONNX models coverage:

ONNX backend tests include below models. The table lists current status on Mxnet: