Codegen Operator List

View Source On Gitee

This article lists the operators supported by MindSpore Lite Codegen.

Operation
 

CPU
FP32

CPU
Int8

CMSIS
Int8

TensorFlow Lite
operators supported

Caffe Lite
operators supported

Onnx Lite
operators supported

TensorFlow
operators supported

Abs

Abs

Abs

Add

Add

Add, Int8Add

Add, AddV2

AddN

AddN

AvgPool

MeanPooling

Pooling

AveragePool,
GlobalAveragePool,
Int8AveragePool

BatchNorm

BatchNorm

BatchNormalization

BiasAdd

BiasAdd

BiasAdd

Cast

Cast, QUANTIZE,
DEQUANTIZE

Cast

Cast

Ceil

Ceil

Ceil

Concat

Concat

Concat

Concat

ConcatV2

Conv2d

Conv2D

Convolution

Conv,
Int8Conv,
ConvRelu,
Int8ConvRelu

Conv2D

Cos

Cos

Cos

DetectionPostProcess

Custom

Div

Div, RealDiv

Div

Div, RealDiv

Eltwise

Eltwise

Sum, Max[3]

Equal

Equal

Equal

Equal

ExpandDims

ExpandDims

ExpandDims

Floor

flOOR

Floor

FloorDiv

FloorDiv

FloorMod

FloorMod

FullConnection

FullyConnected

InnerProduct

Greater

Greater

Greater

Greater

GreaterEqual

GreaterEqual

GreaterEqual

Less

Less

Less

Less

LessEqual

LessEqual

LessEqual

Log

Log

Log

LogicalAnd

LogicalAnd

And

LogicalAnd

LogicalNot

LogicalNot

Not

LogicalOr

LogicalOr

Or

MatMul

MatMul

MatMul

Maximum

Maximum

Maximum

MaxPool

MaxPooling

Pooling

MaxPool,
GlobalMaxPool

Minimum

Minimum

Min

Minimum

Mul

Mul

Mul

Mul

Neg

Neg

Neg

NotEqual

NotEqual

NotEqual

ReLU

Relu

ReLU

Relu

Relu

ReLU6

Relu6

ReLU6

Clip[1]

Relu6

Reshape

Reshape

Reshape

Reshape,Flatten

Reshape

Resize

ResizeBilinear,
NearestNeighbor

Interp

Round

Round

Round

Round

Rsqrt

Rsqrt

Sigmoid

Logistic

Sigmoid

Sigmoid

Sigmoid

Sin

Sin

Sin

Softmax

Softmax

Softmax

Softmax

Sqrt

Sqrt

Sqrt

Square

Square

SquaredDifference

SquaredDifference

Squeeze

Squeeze

Squeeze

Squeeze

Sub

Sub

Sub

Sub

[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.