开源项目 > 其他开源 > 图书/手册/教程 && 人工智能 > 机器学习/深度学习 &&

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

克隆/下载

operator_list_codegen.md 11.72 KB

# Codegen算子支持

[![查看源文件](https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/r1.8/resource/_static/logo_source.png)](https://gitee.com/mindspore/docs/blob/r1.8/docs/lite/docs/source_zh_cn/operator_list_codegen.md)

本文列举MindSpore Lite Codegen支持的算子。

| 操作名 <br/>&nbsp;   | CPU<br/>FP32 | CPU<br/>Int8 | CMSIS<br/>Int8  | 支持的TensorFlow Lite算子    | 支持的Caffe Lite算子  | 支持的Onnx Lite算子                          |支持的TensorFlow算子                         |
|-----------------------|:--------------:|:--------------:|:-----------------:|---------------------------------|--------------------------|-------------------------------------------------|-------------------------------------------------|
| Abs                   | ✅    |              |                 | Abs                             |                          | Abs                                             |                                                 |
| Add                   | ✅    | ✅    | ✅       | Add                             |                          | Add,<br/>Int8Add                                    | Add, AddV2                                      |
| AddN                  | ✅    |              |                 | AddN                            |                          |                                                 |                                                 |
| AvgPool               | ✅    | ✅    | ✅       | MeanPooling                     | Pooling                  | AveragePool,<br/>GlobalAveragePool,<br/>Int8AveragePool |                                                 |
| BatchNorm             |              | ✅    | ✅       |                                 | BatchNorm                | BatchNormalization                              |                                                 |
| BiasAdd               | ✅    |              |                 |                                 |                          | BiasAdd                                         | BiasAdd                                         |
| Cast                  | ✅    | ✅    | ✅       | Cast, QUANTIZE,<br/>DEQUANTIZE      |                          | Cast                                            | Cast                                            |
| Ceil                  | ✅    |              |                 | Ceil                            |                          | Ceil                                            |                                                 |
| Concat                | ✅    | ✅    | ✅       | Concat                          | Concat                   | Concat                                          | ConcatV2                                        |
| Conv2d                | ✅    | ✅    | ✅       | Conv2D                          | Convolution              | Conv, Int8Conv,<br/>ConvRelu,<br/>Int8ConvRelu          | Conv2D                                          |
| Cos                   | ✅    |              |                 | Cos                             |                          | Cos                                             |                                                 |
| DetectionPostProcess  |              | ✅    |                 | Custom                          |                          |                                                 |                                                 |
| Div                   | ✅    | ✅    | ✅       | Div, RealDiv                    |                          | Div                                             | Div, RealDiv                                    |
| Eltwise               | ✅    |              |                 |                                 | Eltwise                  | Sum, Max<sup>[3]</sup>                          |                                                 |
| 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,<br/>GlobalMaxPool                          |                                                 |
| Minimum               | ✅    |              |                 | Minimum                         |                          | Min                                             | Minimum                                         |
| Mul                   | ✅    | ✅    | ✅       | Mul                             |                          | Mul                                             | Mul                                             |
| Neg                   | ✅    |              |                 | Neg                             |                          | Neg                                             |                                                 |
| NotEqual              | ✅    |              |                 | NotEqual                        |                          |                                                 |NotEqual                                         |
| ReLU                  | ✅    | ✅    | ✅       | Relu                            | ReLU                     | Relu                                            | Relu                                            |
| ReLU6                 | ✅    | ✅    | ✅       | Relu6                           | ReLU6                    | Clip<sup>[1]</sup>                              | Relu6                                           |
| Reshape               | ✅    | ✅    | ✅       | Reshape                         | Reshape                  | Reshape,Flatten                                 | Reshape                                         |
| Resize                |              | ✅    |                 | ResizeBilinear,<br/>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。

一键复制编辑原始数据按行查看历史

提交于 2023-07-21 16:52 . modify the md links 1.8

Codegen算子支持

本文列举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。

1

https://gitee.com/mindspore/docs.git

git@gitee.com:mindspore/docs.git

mindspore

docs

docs

r1.8