Codegen Operator List
Linux
Ascend
Device
Inference
Beginner
Intermediate
Expert
This article lists the operators supported by MindSpore Lite Codegen.
Operation |
CPU |
CPU |
CMSIS |
TensorFlow Lite |
Caffe Lite |
Onnx Lite |
TensorFlow |
---|---|---|---|---|---|---|---|
Abs |
✅ |
Abs |
Abs |
||||
Add |
✅ |
✅ |
✅ |
Add |
Add, Int8Add |
Add, AddV2 |
|
AddN |
✅ |
AddN |
|||||
AvgPool |
✅ |
✅ |
✅ |
MeanPooling |
Pooling |
AveragePool, |
|
BatchNorm |
✅ |
✅ |
BatchNorm |
BatchNormalization |
|||
BiasAdd |
✅ |
BiasAdd |
BiasAdd |
||||
Cast |
✅ |
✅ |
✅ |
Cast, QUANTIZE, |
Cast |
Cast |
|
Ceil |
✅ |
Ceil |
Ceil |
||||
Concat |
✅ |
✅ |
✅ |
Concat |
Concat |
Concat |
ConcatV2 |
Conv2d |
✅ |
✅ |
✅ |
Conv2D |
Convolution |
Conv, |
Conv2D |
Cos |
✅ |
Cos |
Cos |
||||
DetectionPostProcess |
✅ |
Custom |
|||||
Div |
✅ |
✅ |
✅ |
Div, RealDiv |
Div |
Div, RealDiv |
|
Eltwise |
✅ |
Eltwise |
Sum, Max[3] |
||||
Equal |
✅ |
Equal |
Equal |
Equal |
|||
ExpandDims |
✅ |
ExpandDims |
ExpandDims |
||||
Floor |
✅ |
flOOR |
Floor |
||||
FloorDiv |
✅ |
FloorDiv |
|||||
FloorMod |
✅ |
FloorMod |
|||||
FullConnection |
✅ |
✅ |
✅ |
FullyConnected |
InnerProduct |
||
Greater |
✅ |
Greater |
Greater |
Greater |
|||
GreaterEqual |
✅ |
GreaterEqual |
GreaterEqual |
||||
Less |
✅ |
Less |
Less |
Less |
|||
LessEqual |
✅ |
LessEqual |
LessEqual |
||||
Log |
✅ |
Log |
Log |
||||
LogicalAnd |
✅ |
LogicalAnd |
And |
LogicalAnd |
|||
LogicalNot |
✅ |
LogicalNot |
Not |
||||
LogicalOr |
✅ |
LogicalOr |
Or |
||||
MatMul |
✅ |
✅ |
MatMul |
MatMul |
|||
Maximum |
✅ |
Maximum |
Maximum |
||||
MaxPool |
✅ |
✅ |
✅ |
MaxPooling |
Pooling |
MaxPool, |
|
Minimum |
✅ |
Minimum |
Min |
Minimum |
|||
Mul |
✅ |
✅ |
✅ |
Mul |
Mul |
Mul |
|
Neg |
✅ |
Neg |
Neg |
||||
NotEqual |
✅ |
NotEqual |
NotEqual |
||||
ReLU |
✅ |
✅ |
✅ |
Relu |
ReLU |
Relu |
Relu |
ReLU6 |
✅ |
✅ |
✅ |
Relu6 |
ReLU6 |
Clip[1] |
Relu6 |
Reshape |
✅ |
✅ |
✅ |
Reshape |
Reshape |
Reshape,Flatten |
Reshape |
Resize |
✅ |
ResizeBilinear, |
Interp |
||||
Round |
✅ |
Round |
Round |
Round |
|||
Rsqrt |
✅ |
Rsqrt |
|||||
Sigmoid |
✅ |
✅ |
✅ |
Logistic |
Sigmoid |
Sigmoid |
Sigmoid |
Sin |
✅ |
Sin |
Sin |
||||
Softmax |
✅ |
✅ |
✅ |
Softmax |
Softmax |
Softmax |
|
Sqrt |
✅ |
Sqrt |
Sqrt |
||||
Square |
✅ |
Square |
|||||
SquaredDifference |
✅ |
SquaredDifference |
|||||
Squeeze |
✅ |
Squeeze |
Squeeze |
Squeeze |
|||
Sub |
✅ |
✅ |
✅ |
Sub |
Sub |
Sub |
[1] Clip: Only support converting clip(0, 6) to Relu6.
[2] Pow: Only support the form where the exponent is a single constant.
[3] Sum and Max: Only support 2 inputs.