MindSpore Lite支持的TensorFlow Lite算子列表

查看源文件

MindSpore Lite算子名称

对应的TensorFlow Lite算子

算子规格

Abs

Abs

-

Activation

Activation, ReLU, ReLU6, PReLU,
LeakyReLU, Tanh, HardSwish, Logistic

-

Adam

Adam

-

AddFusion

Add

-

AddN

AddN

-

ApplyMomentum

ApplyMomentum

-

Assign

Assign

-

ArgmaxFusion

Argmax

-

ArgminFusion

Argmin

-

AvgPoolFusion

MeanPooling

-

BatchToSpace

BatchToSpace

-

BatchToSpaceND

BatchToSpaceND

-

BinaryCrossEntropy

BinaryCrossEntropy

-

BroadcastTo

BroadcastTo

-

Cast

Cast,
QUANTIZE,
DEQUANTIZE

-

Ceil

Ceil

-

Clip

Clip

-

Concat

Concat

-

Conv2DFusion

Conv2D

-

Conv2dTransposeFusion

DeConv2D

-

Cos

Cos

-

CustomExtractFeatures

ExtractFeatures

-

CustomNormalize

Normalize

-

CustomPredict

Predict

-

DepthToSpace

DepthToSpace

-

DetectionPostProcess

Custom

-

DivFusion

Div, RealDiv

-

Dropout

Dropout

-

Equal

Equal

-

Erf

Erf

-

ExpFusion

Exp

-

ExpandDims

ExpandDims

-

Fill

Fill

-

Floor

Floor

-

FloorDiv

FloorDiv

-

FloorMod

FloorMod

-

FullConnection

FullyConnected

-

FusedBatchNorm

FusedBatchNorm

-

GatherNd

GatherND

-

Gather

Gather

-

Greater

Greater

-

GreaterEqual

GreaterEqual

-

HashtableLookup

HashtableLookup

-

InstanceNorm

InstanceNorm

-

IsFinite

IsFinite

-

LeakyReLU

LeakyRelu

-

Less

Less

-

LessEqual

LessEqual

-

LRN

LocalResponseNorm

-

Log

Log

-

Log1p

Log1p

-

LogicalAnd

LogicalAnd

-

LogicalNot

LogicalNot

-

LogicalOr

LogicalOr

-

LogSoftmax

LogSoftmax

-

LshProjection

LshProjection

-

MatMulFusion

BatchMatMul

-

Maximum

Maximum

-

MaxPoolFusion

MaxPooling

-

Minimum

Minimum

-

Mod

Mod

-

MulFusion

Mul

-

Neg

Neg

-

NotEqual

NotEqual

-

NonMaxSuppression

NonMaxSuppression

-

NonZero

NonZero

-

OneHot

OneHot

-

OnesLike

OnesLike

-

PadFusion

Pad, MirrorPad, PadV2

-

PowFusion

Pow

-

PReLUFusion

PRELU

-

RandomNormal

RandomNormal

-

Range

Range

-

Rank

Rank

-

ReduceFusion

Sum, Mean, ReduceMax, ReduceMin, ReduceProd

-

Reshape

Reshape

-

Resize

ResizeBilinear,
NearestNeighbor

-

ReverseV2

reverse

-

ReverseSequence

ReverseSequence

-

Round

Round

-

Rsqrt

Rsqrt

-

ScatterNd

ScatterNd

-

ScatterNdUpdate

ScatterNdUpdate

-

SGD

SGD

-

Shape

Shape

-

Sin

Sin

-

SliceFusion

Slice

-

SkipGram

SkipGram

-

Softmax

Softmax

-

SpaceToBatch

SpaceToBatch

仅支持4维输入张量。

SpaceToBatchND

SpaceToBatchND

-

SpaceToDepth

SpaceToDepth

-

SparseToDense

SparseToDense

-

Split

Split, SplitV

-

Sqrt

Sqrt

-

Square

Square

-

SquaredDifference

SquaredDifference

-

Squeeze

Squeeze

-

StridedSlice

StridedSlice

-

Stack

Stack

-

SubFusion

Sub

-

TensorScatterAdd

TensorScatterAdd

-

TileFusion

Tile

-

TopKFusion

TopKV2

-

Transpose

Transpose

-

Unique

Unique

-

Unstack

Unstack

-

Where

Where

-

ZerosLike

ZerosLike

-