代码拉取完成,页面将自动刷新
本文列举MindSpore Lite Codegen支持的算子。
操作名 |
CPU FP32 |
CPU Int8 |
CMSIS Int8 |
支持的TensorFlow Lite算子 | 支持的Caffe Lite算子 | 支持的Onnx Lite算子 | 支持的TensorFlow算子 |
---|---|---|---|---|---|---|---|
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:仅支持将clip(0, 6)转换为Relu6。
[2] Pow:仅支持指数为单个常数。
[3] Sum与Max:仅支持输入个数为2。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。