List of TensorFlow Operators Supported by MindSpore Lite
MindSpore Lite Operator Names |
Corresponding TensorFlow Operators |
---|---|
Abs |
Abs |
Activation |
Activation, Elu, Relu, Relu6, Sigmoid, Tanh, Selu, LeakyRelu, Softplus |
Adam |
Adam |
AddFusion |
Add, AddV2 |
All |
All |
ApplyMomentum |
ApplyMomentum |
Assert |
Assert |
Assign |
Assign |
ArgmaxFusion |
ArgMax |
ArgminFusion |
ArgMin |
AvgPoolFusion |
AvgPool |
BatchToSpace |
BatchToSpace |
BatchToSpaceND |
BatchToSpaceND |
BiasAdd |
BiasAdd |
BinaryCrossEntropy |
BinaryCrossEntropy |
BroadcastTo |
BroadcastTo |
Cast |
Cast |
Ceil |
Ceil |
Clip |
Clip |
Concat |
ConcatV2 |
Conv2DFusion |
Conv2D |
Conv2dTransposeFusion |
Conv2DBackpropInput |
Cos |
Cos |
CropAndResize |
CropAndResize |
CumSum |
Cumsum |
DepthToSpace |
DepthToSpace |
DivFusion |
Div, RealDiv |
Dropout |
Dropout |
Elu |
NonMaxSuppressionV3 |
Equal |
Equal |
Erf |
Erf |
ExpFusion |
Exp |
ExpandDims |
ExpandDims |
Fill |
Fill |
Floor |
Floor |
FloorDiv |
FloorDiv |
FloorMod |
FloorMod |
FusedBatchNorm |
FusedBatchNorm, |
GatherNd |
GatherNd |
Gather |
GatherV2 |
Greater |
Greater |
GreaterEqual |
GreaterEqual |
InvertPermutation |
InvertPermutation |
IsFinite |
IsFinite |
LeakyReLU |
LeakyRelu |
Less |
Less |
LessEqual |
LessEqual |
Log |
Log |
Log1p |
Log1p |
LogicalAnd |
LogicalAnd |
LogicalNot |
LogicalNot |
LogicalOr |
LogicalOr |
MatMulFusion |
MatMul, |
Maximum |
Maximum |
MaxPoolFusion |
MaxPool |
Merge |
Merge |
Minimum |
Minimum |
Mod |
Mod |
MulFusion |
Mul |
Neg |
Neg |
NotEqual |
NotEqual |
NonMaxSuppression |
NonMaxSuppression |
NonZero |
NonZero |
OneHot |
OneHot |
OnesLike |
OnesLike |
PadFusion |
MirrorPad, Pad, PadV2 |
PowFusion |
Pow |
RaggedRange |
RaggedRange |
RandomNormal |
RandomNormal |
RandomStandardNormal |
RandomStandardNormal |
Range |
Range |
Rank |
Rank |
ReduceFusion |
Sum, Max, Min, Mean, Prod, All |
Reshape |
Reshape |
Resize |
ResizeBilinear, |
ReverseV2 |
ReverseV2 |
ReverseSequence |
ReverseSequence |
Round |
Round |
Rsqrt |
Rsqrt |
Select |
Select |
Selu |
Selu |
SGD |
SGD |
Shape |
Shape |
Sin |
Sin |
Size |
Size |
SliceFusion |
Slice |
Softmax |
Softmax |
Softplus |
Softplus |
SpaceToBatchND |
SpaceToBatchND |
Split |
Split, SplitV |
Sqrt |
Sqrt |
Square |
Square |
SquaredDifference |
SquaredDifference |
Squeeze |
Squeeze |
StridedSlice |
StridedSlice |
Stack |
Pack |
SubFusion |
Sub |
Switch |
Switch |
TensorListFromTensor |
TensorListFromTensor |
TensorListGetItem |
TensorListGetItem |
TensorListReserve |
TensorListReserve |
TensorListSetItem |
TensorListSetItem |
TensorListStack |
TensorListStack |
TensorScatterAdd |
TensorScatterAdd |
TileFusion |
Tile |
TopKFusion |
TopKV2 |
Transpose |
Transpose |
UnsortedSegmentSum |
UnsortedSegmentSum |
Where |
Where |
ZerosLike |
ZerosLike |
Other operators supported by the conversion tool |
Dropout, Enter, |
Converter too supports operators that are not required to be explicitly implemented. Typically, such operators are optimized away in conversion tools—either fused or replaced with other operators.