MindSpore Lite算子支持

Linux Ascend 端侧 推理应用 初级 中级 高级

操作名
 
CPU
FP16
CPU
FP32
CPU
Int8
CPU
UInt8
GPU
FP16
GPU
FP32
NPU
 
支持的TensorFlow
Lite算子
支持的Caffe
Lite算子
支持的Onnx
Lite算子
支持的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:仅支持将clip(0, 6)转换为Relu6。

[2] Pow:仅支持指数为单个常数。

[3] Sum与Max:仅支持输入个数为2。