简介
AI CPU算子的实现包括两部分:
算子类声明:在头文件中进行算子类的声明,自定义算子类需要继承CpuKernel基类。
计算逻辑实现:在源文件中重写算子类中的Compute函数,实现算子计算逻辑。
下文将对这两部分的实现做具体说明。
如果您想获取AI CPU算子的实现样例,请参考
表1
。
父主题:
算子代码实现(AI CPU)