算子支持

查看源文件

√勾选的项为MindSpore Lite所支持的算子。

操作名

CPU
FP16

CPU
FP32

CPU
Int8

CPU
UInt8

GPU
FP16

GPU
FP32

支持的Tensorflow
Lite op

支持的Caffe
Lite op

支持的Onnx
Lite op

Abs

Abs

Abs

Add

Add

Add

AddN

AddN

Argmax

Argmax

ArgMax

ArgMax

Argmin

Argmin

AvgPool

MeanPooling

Pooling

AveragePool

BatchNorm

BatchNorm

BatchNormalization

BatchToSpace

BatchToSpace, BatchToSpaceND

BiasAdd

BiasAdd

Broadcast

BroadcastTo

Expand

Cast

Cast, DEQUANTIZE*

Cast

Ceil

Ceil

Ceil

Concat

Concat

Concat

Concat

Conv2d

Conv2D

Convolution

Conv

Conv2dTranspose

DeConv2D

Deconvolution

ConvTranspose

Cos

Cos

Cos

Crop

Crop

DeDepthwiseConv2D

Deconvolution

ConvTranspose

DepthToSpace

DepthToSpace

DepthToSpace

DepthwiseConv2dNative

DepthwiseConv2D

Convolution

Convolution

Div

Div, RealDiv

Div

Eltwise

Eltwise

Elu

Elu

Elu

Equal

Equal

Equal

Exp

Exp

Exp

ExpandDims

Fill

Fill

Flatten

Flatten

Floor

flOOR

Floor

FloorDiv

FloorDiv

FloorMod

FloorMod

FullConnection

FullyConnected

InnerProduct

GatherNd

GatherND

GatherV2

Gather

Gather

Greater

Greater

Greater

GreaterEqual

GreaterEqual

Hswish

HardSwish

LeakyReLU

LeakyRelu

LeakyRelu

Less

Less

Less

LessEqual

LessEqual

LRN

LocalResponseNorm

Lrn

Log

Log

Log

LogicalAnd

LogicalAnd

LogicalNot

LogicalNot

LogicalOr

LogicalOr

LSTM

MatMul

MatMul

Maximum

Maximum

Max

MaxPool

MaxPooling

Pooling

MaxPool

Minimum

Minimum

Min

Mul

Mul

Mul

NotEqual

NotEqual

OneHot

OneHot

Pad

Pad

Pad

Pow

Pow

Power

Power

PReLU

PReLU

Range

Range

Rank

Rank

ReduceMax

ReduceMax

ReduceMax

ReduceMean

Mean

ReduceMean

ReduceMin

ReduceMin

ReduceMin

ReduceProd

ReduceProd

ReduceSum

Sum

ReduceSum

ReduceSumSquare

ReLU

Relu

ReLU

Relu

ReLU6

Relu6

ReLU6

Clip*

Reshape

Reshape

Reshape

Reshape,Flatten

Resize

ResizeBilinear, NearestNeighbor

Interp

Reverse

reverse

ReverseSequence

ReverseSequence

Round

Round

Rsqrt

Rsqrt

Scale

Scale

ScatterNd

ScatterNd

Shape

Shape

Shape

Sigmoid

Logistic

Sigmoid

Sigmoid

Sin

Sin

Sin

Slice

Slice

Slice

Softmax

Softmax

Softmax

Softmax

SpaceToBatch

SpaceToBatchND

SpaceToBatchND

SpaceToDepth

SpaceToDepth

SpaceToDepth

SparseToDense

SpareToDense

Split

Split, SplitV

Sqrt

Sqrt

Sqrt

Square

Square

SquaredDifference

SquaredDifference

Squeeze

Squeeze

Squeeze

StridedSlice

StridedSlice

Stack

Stack

Sub

Sub

Sub

Tanh

Tanh

TanH

Tile

Tile

Tile

TopK

TopKV2

Transpose

Transpose

Permute

Transpose

Unique

Unique

Unsqueeze

Unsqueeze

Unstack

Unstack

Where

Where

ZerosLike

ZerosLike

  • Clip: 仅支持将clip(0, 6)转换为Relu6.

  • DEQUANTIZE: 仅支持将fp16转换为fp32.