更新时间:2021-03-18 GMT+08:00
分享

实现流程

TE算子代码通过Python语言开发,实现流程如图1所示。

图1 TE自定义算子实现流程图
  • 支持的自定义算子的输入数据类型为: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”
分享:

    相关文档

    相关产品