更新时间:2021-03-18 GMT+08:00
实现流程
TE算子代码通过Python语言开发,实现流程如图1所示。
- 支持的自定义算子的输入数据类型为:float16, int8, int16, int32, uint8, uint16, bool。
- 不同计算操作支持的数据类型不同,详细请参见《TE API参考》。
- TE API同时支持float16与float32数据类型,但OMG进行模型转换的时候会将float32数据类型转换成float16,所以当前版本在进行自定义算子开发时,不支持使用float32的数据类型。
- TE提供了一些已经自定义实现的算子样例代码,用户可参考或直接使用,代码存储路径为DDK安装目录下的“ddk/site-packages/topi-0.4.0.egg/topi/cce”。
父主题: 算子代码实现