简介
AI CPU算子的实现包括两部分:
- 算子类声明:在头文件中进行算子类的声明,自定义算子类需要继承CpuKernel基类。
- 计算逻辑实现:在源文件中重写算子类中的Compute函数,实现算子计算逻辑。
下文将对这两部分的实现做具体说明。
如果您想获取AI CPU算子的实现样例,请参考表1。
父主题: 算子代码实现(AI CPU)
AI CPU算子的实现包括两部分:
下文将对这两部分的实现做具体说明。
如果您想获取AI CPU算子的实现样例,请参考表1。
当前产品无相关内容
未找到相关内容,请尝试其他搜索词