THIS IS A TEST INSTANCE. ALL YOUR CHANGES WILL BE LOST!!!!
...
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
...
| Operator | Test Coverage | Mxnet (Import) | Mxnet (Export) | ||||
|---|---|---|---|---|---|---|---|
| Abs | Yes | OK | OK | ||||
| Acos | Yes | OK | OK, In review | ||||
| Add | Yes | OK | OK | ||||
| And | Yes | OK | OK | ||||
| ArgMax | Yes | OK | OK | ||||
| ArgMin | Yes | OK | OK | ||||
| Asin | Yes | OK | OK, In review | ||||
| Atan | Yes | OK | OkOK, In review | ||||
| AveragePool | Yes | OK | OK | ||||
| BatchNormalization | Yes | OK | OK | ||||
| Cast | Yes | OK | OK | ||||
| Ceil | Yes | OK | OK | ||||
| Clip | Yes | OK | OK | ||||
| Concat | Yes | OK | OK | ||||
| Constant | Yes | OK | |||||
| Conv | Yes | OK | OK | ||||
| ConvTranspose | Yes | OK | OK | ||||
| Cos | OK, In review | Yes | OK | OK | |||
| Crop | OK | ||||||
| DepthToSpace | Yes | OK | OK | DepthToSpace | NOT OK, under development | ||
| Div | Yes | OK | OK | ||||
| Dropout | Yes | OK | OK | ||||
| Elu | Yes | OK | OK | ||||
| Equal | Yes | OK | OK | ||||
| Exp | Yes | OK | OK | ||||
| Flatten | Yes | PARTIAL, only supports default axis=1 | OK | ||||
| Floor | Yes | OK | OK | ||||
| GRU | NOT OK , under development | ||||||
| Gather | Yes | OK | |||||
| Gemm | Yes | OK | OK | ||||
| GlobalAveragePool | Yes | OK | OK | ||||
| GlobalLpPool | Yes | OK, In review | OK | ||||
| GlobalMaxPool | Yes | OK | OK | ||||
| Greater | Yes | OK | OK | ||||
| HardSigmoid | Yes | OK | OK | ||||
| Hardmax | Yes | OK, In review | NOT OK | ||||
| Identity | NOT OK | OK | |||||
| If | |||||||
| InstanceNormalization | Yes | OK, In review | OK | ||||
| LRN | Yes | OK | OK | ||||
| LSTM | NOT OK, under development | ||||||
| LeakyRelu | Yes | OK | OK | ||||
| Less | Yes | OK | OK | ||||
| Log | Yes | OK | OK | ||||
| LogSoftMax | Yes | OK | OK | ||||
| Loop | |||||||
| LpNormalization | OK, In review | Yes | OK | ||||
| LpPool | Yes | OK, In review | |||||
| MatMul | Yes | OK | OKOK, In review | ||||
| Max | Yes | OK | OK | ||||
| MaxPool | Yes | OK | OK | ||||
| MaxRoiPool | Yes | OK, In review | OK | ||||
| Mean | Yes | OK | OK | ||||
| Min | Yes | OK | OK | ||||
| Mul | Yes | OK | OK | ||||
| Multinomial | Yes | OK | OK, In review | ||||
| Neg | Yes | OK | OK | ||||
| Not | Yes | OK | OK | ||||
| Or | Yes | OK | OK | ||||
| PRelu | Yes | OK | OK | ||||
| Pad | Yes | OK | OK | ||||
| Pow | Yes | Partial, only supports default axis=1 | OK | ||||
| RNN | NOT OK, under development | ||||||
| RandomNormal | Yes | OK | OK | ||||
| RandomNormalLike | Yes | OK | OK | ||||
| RandomUniform | Yes | OK | OK | ||||
| RandomUniformLike | Yes | OK | OK | ||||
| Reciprocal | Yes | OK | OK | ||||
| ReduceL1 | Yes | OK | OK | ||||
| ReduceL2 | Yes | OK | OK | ||||
| ReduceLogSum | Yes | OK | NOT OK | ||||
| ReduceLogSumExp | Yes | OK | NOT OK | ||||
| ReduceMax | Yes | OK | OK | ||||
| ReduceMean | Yes | OK | OK | ||||
| ReduceMin | Yes | OK | OK | ||||
| ReduceProd | Yes | OK | OK | ||||
| ReduceSum | Yes | OK | OK | ||||
| ReduceSumSquare | Yes | OK | NOT OK | ||||
| Relu | Yes | OK | OK | ||||
| Reshape | Yes | OK | OK | ||||
| Selu | Yes | NOT OK, under development | OK | ||||
| Shape | OK, In review | OK | |||||
| Sigmoid | Yes | OK | OK | ||||
| Sin | Yes | OK, In review | OK | ||||
| Size | Yes | OK | OK, In review | ||||
| Slice | Yes | Partial, supports axis=1 | OK | ||||
| Softmax | Yes | OK | OK | ||||
| Softplus | Yes | OK | OK | ||||
| Softsign | Yes | OK | OK | ||||
| SpaceToDepth | Yes | OK | NOT OK, under development | ||||
| Split | Yes | OK | OK | ||||
| Sqrt | Yes | OK | OK | ||||
| Squeeze | Yes | OK | OK | ||||
| Sub | Yes | OK | OK | ||||
| Sum | Yes | OK | OK | ||||
| Tan | Yes | OK | OKOK, In review | ||||
| Tanh | Yes | OK | OK | ||||
| Tile | Yes | NOT OK, under development | NOT OK, under development | ||||
| TopK | Yes | OK, in review | OK, in In review | ||||
| Transpose | Yes | OK | OK | ||||
| Unsqueeze | Yes | OK, In review | OK | ||||
| Upsample | Yes | OK, in review | OK, in In review | ||||
| Xor | Yes | OK | OK | ||||
| experimental ATen | NOT OK | ||||||
| experimental Affine | NOT OK | ||||||
| experimental ConstantFill | NOT OK | experimental Crop | NOT OK | ||||
| experimental GRUUnit | NOT OK | ||||||
| experimental GivenTensorFill | NOT OK | ||||||
| experimental If | NOT OK | experimental ImageScaler | Yes | OK, In review | |||
| experimental Loop | NOT OK | experimental LoopIndex | NOT OK | ||||
| experimental MeanVarianceNormalization | NOT OK | ||||||
| experimental ParametricSoftplus | NOT OK | ||||||
| experimental Scale | NOT OK | ||||||
| experimental ScaledTanh | NOT OK | ||||||
| experimental ThresholdedRelu | NOT OK |
ONNX models coverage:
ONNX backend tests include below models. The table lists current status on Mxnet:
| ONNX Models | Mxnet Support |
|---|---|
| bvlc_alexnet | OK |
| densenet121 | OK |
| inception_v1 | OK, In review |
| inception_v2 | OK, In review |
| resnet50 | OK |
| shufflenet | OK, In review |
| squeezenet | OK |
| vgg16 | OK |
| vgg19 | OK |
| bvlc_googlenet | OK |
| bvlc_caffenet | OK |
| bvlc-rcnn-ilsvrc13 | OK |
...