MindSpore Lite Operator List
Linux
On Device
Inference Application
Beginner
Intermediate
Expert
Operation |
CPU |
CPU |
CPU |
CPU |
GPU |
GPU |
NPU |
TensorFlow |
Caffe |
Onnx |
TensorFlow |
---|---|---|---|---|---|---|---|---|---|---|---|
Abs |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Abs |
Abs |
||
Add |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Add |
Add, Int8Add |
Add, AddV2 |
|
AddN |
Supported |
AddN |
|||||||||
Assert |
Supported |
Assert |
|||||||||
Argmax |
Supported |
Supported |
Supported |
Supported |
Supported |
Argmax |
ArgMax |
ArgMax |
|||
Argmin |
Supported |
Supported |
Supported |
Supported |
Supported |
Argmin |
|||||
AvgPool |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
MeanPooling |
Pooling |
AveragePool, GlobalAveragePool, Int8AveragePool |
|
BatchNorm |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
BatchNorm |
BatchNormalization |
|||
BatchToSpace |
Supported |
Supported |
Supported |
Supported |
Supported |
BatchToSpace, BatchToSpaceND |
|||||
BiasAdd |
Supported |
Supported |
Supported |
Supported |
Supported |
BiasAdd |
BiasAdd |
||||
Broadcast |
Supported |
BroadcastTo |
Expand |
||||||||
Cast |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Cast, QUANTIZE, DEQUANTIZE |
Cast |
Cast |
|
Ceil |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Ceil |
Ceil |
||
Concat |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Concat |
Concat |
Concat |
ConcatV2 |
ConstantOfShape |
Supported |
ConstantOfShape |
|||||||||
Conv2d |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Conv2D |
Convolution |
Conv, Int8Conv, ConvRelu, Int8ConvRelu |
Conv2D |
Conv2dTranspose |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
DeConv2D |
Deconvolution |
ConvTranspose |
|
Cos |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Cos |
Cos |
||
Crop |
Supported |
Supported |
Supported |
Supported |
Crop |
||||||
CustomExtractFeatures |
Supported |
ExtractFeatures |
|||||||||
CustomNormalize |
Supported |
Normalize |
|||||||||
CustomPredict |
Supported |
Predict |
|||||||||
DeDepthwiseConv2D |
Supported |
Supported |
Supported |
Deconvolution |
|||||||
DepthToSpace |
Supported |
Supported |
Supported |
Supported |
Supported |
DepthToSpace |
DepthToSpace |
||||
DepthwiseConv2dNative |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
DepthwiseConv2D |
Convolution |
||
DetectionPostProcess |
Supported |
Supported |
Supported |
Custom |
|||||||
Div |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Div, RealDiv |
Div |
Div, RealDiv |
|
Eltwise |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Eltwise |
Sum, Max[3] |
||
Elu |
Supported |
Elu |
Elu, NonMaxSuppression |
||||||||
Equal |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Equal |
Equal |
Equal |
|
Exp |
Supported |
Supported |
Supported |
Exp |
Exp |
Exp |
|||||
ExpandDims |
Supported |
Supported |
Supported |
ExpandDims |
ExpandDims |
||||||
Fill |
Supported |
Fill |
|||||||||
Flatten |
Supported |
Flatten |
|||||||||
Floor |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
flOOR |
Floor |
||
FloorDiv |
Supported |
Supported |
Supported |
Supported |
Supported |
FloorDiv |
|||||
FloorMod |
Supported |
Supported |
Supported |
Supported |
Supported |
FloorMod |
|||||
FullConnection |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
FullyConnected |
InnerProduct |
|||
FusedBatchNorm |
Supported |
Supported |
Supported |
Supported |
Supported |
FusedBatchNorm |
|||||
GatherNd |
Supported |
Supported |
Supported |
Supported |
Supported |
GatherND |
|||||
Gather |
Supported |
Supported |
Supported |
Supported |
Supported |
Gather |
Gather |
GatherV2 |
|||
Greater |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Greater |
Greater |
Greater |
|
GreaterEqual |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
GreaterEqual |
GreaterEqual |
||
HashtableLookup |
Supported |
HashtableLookup |
|||||||||
Hswish |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
HardSwish |
|||
InstanceNorm |
Supported |
InstanceNorm |
|||||||||
L2Norm |
Supported |
L2_NORMALIZATION |
|||||||||
LeakyReLU |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
LeakyRelu |
LeakyRelu |
||
Less |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Less |
Less |
Less |
|
LessEqual |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
LessEqual |
LessEqual |
||
LRN |
Supported |
LocalResponseNorm |
Lrn, LRN |
||||||||
Log |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Log |
Log |
||
LogicalAnd |
Supported |
Supported |
Supported |
Supported |
Supported |
LogicalAnd |
And |
LogicalAnd |
|||
LogicalNot |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
LogicalNot |
Not |
||
LogicalOr |
Supported |
Supported |
Supported |
Supported |
Supported |
LogicalOr |
Or |
||||
LshProjection |
Supported |
LshProjection |
|||||||||
LSTM |
Supported |
LSTM |
|||||||||
MatMul |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
MatMul |
MatMul |
||
Maximum |
Supported |
Supported |
Supported |
Supported |
Supported |
Maximum |
Maximum |
||||
MaxPool |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
MaxPooling |
Pooling |
MaxPool, GlobalMaxPool |
|
Minimum |
Supported |
Supported |
Supported |
Supported |
Supported |
Minimum |
Min |
Minimum |
|||
Mul |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Mul |
Mul |
Mul |
|
Neg |
Supported |
Supported |
Supported |
Supported |
Supported |
Neg |
Neg |
||||
NotEqual |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
NotEqual |
NotEqual |
||
OneHot |
Supported |
Supported |
Supported |
OneHot |
OneHot |
||||||
Pad |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Pad, MirrorPad |
Pad |
||
Pow |
Supported |
Supported |
Supported |
Supported |
Supported |
Pow |
Power |
Pow[2] |
|||
PReLU |
Supported |
Supported |
Supported |
PRELU |
PReLU |
PRelu |
|||||
Range |
Supported |
Range |
Range, RaggedRange |
||||||||
Rank |
Supported |
Rank |
|||||||||
ReduceAll |
Supported |
All |
|||||||||
ReduceASum |
Supported |
Supported |
Supported |
Reduction |
|||||||
ReduceMax |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
ReduceMax |
ReduceMax |
Max |
||
ReduceMean |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Mean |
Reduction |
ReduceMean |
Mean |
|
ReduceMin |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
ReduceMin |
ReduceMin |
Min |
||
ReduceProd |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
ReduceProd |
ReduceProd |
Prod |
||
ReduceSum |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Sum |
Reduction |
ReduceSum |
Sum |
|
ReduceSumSquare |
Supported |
Supported |
Supported |
Supported |
Reduction |
ReduceSumSquare |
|||||
ReLU |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Relu |
ReLU |
Relu |
Relu |
ReLU6 |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Relu6 |
ReLU6 |
Clip[1] |
Relu6 |
Reshape |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Reshape |
Reshape |
Reshape,Flatten |
Reshape |
|
Resize |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
ResizeBilinear, NearestNeighbor |
Interp |
|||
Reverse |
Supported |
reverse |
|||||||||
ReverseSequence |
Supported |
ReverseSequence |
ReverseSequence |
||||||||
Round |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Round |
Round |
Round |
|
Rsqrt |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Rsqrt |
|||
Scale |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Scale |
|||
ScatterNd |
Supported |
ScatterNd |
|||||||||
Shape |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Shape |
Shape |
Shape |
||
Sigmoid |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Logistic |
Sigmoid |
Sigmoid |
Sigmoid |
|
Sin |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Sin |
Sin |
||
Slice |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Slice |
Slice |
Slice |
|
SkipGram |
Supported |
SKipGram |
|||||||||
Softmax |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Softmax |
Softmax |
Softmax |
|
SpaceToBatch |
Supported |
Supported |
Supported |
Supported |
Supported |
SpaceToBatch |
|||||
SpaceToBatchND |
Supported |
Supported |
Supported |
Supported |
Supported |
SpaceToBatchND |
|||||
SpaceToDepth |
Supported |
Supported |
Supported |
SpaceToDepth |
SpaceToDepth |
||||||
SparseToDense |
Supported |
Supported |
Supported |
SpareToDense |
|||||||
Split |
Supported |
Supported |
Supported |
Supported |
Supported |
Split, SplitV |
Split |
Split, SplitV |
|||
Sqrt |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Sqrt |
Sqrt |
||
Square |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Square |
|||
SquaredDifference |
Supported |
Supported |
Supported |
Supported |
Supported |
SquaredDifference |
|||||
Squeeze |
Supported |
Supported |
Supported |
Supported |
Supported |
Squeeze |
Squeeze |
Squeeze |
|||
StridedSlice |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
StridedSlice |
StridedSlice |
|||
Stack |
Supported |
Supported |
Supported |
Supported |
Stack |
Pack |
|||||
Sub |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Supported |
Sub |
Sub |
Sub |
|
Tanh |
Supported |
Supported |
Supported |
Supported |
Tanh |
TanH |
Tanh, Sign |
Tanh |
|||
TensorListFromTensor |
Supported |
TensorListFromTensor |
|||||||||
TensorListGetItem |
Supported |
TensorListGetItem |
|||||||||
TensorListReserve |
Supported |
TensorListReserve |
|||||||||
TensorListSetItem |
Supported |
TensorListSetItem |
|||||||||
TensorListStack |
Supported |
TensorListStack |
|||||||||
Tile |
Supported |
Tile |
Tile |
Tile |
Tile |
||||||
TopK |
Supported |
Supported |
Supported |
TopKV2 |
TopK |
||||||
Transpose |
Supported |
Supported |
Supported |
Supported |
Supported |
Transpose |
Permute |
Transpose |
Transpose |
||
Unique |
Supported |
Unique |
|||||||||
Unsqueeze |
Supported |
Supported |
Supported |
Supported |
Unsqueeze |
||||||
Unstack |
Supported |
Unstack |
|||||||||
Where |
Supported |
Where |
|||||||||
While |
Supported |
While, StatelessWhile |
|||||||||
ZerosLike |
Supported |
ZerosLike |
[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.