下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

简介

AI CPU算子的实现包括两部分:

  • 算子类声明:在头文件中进行算子类的声明,自定义算子类需要继承CpuKernel基类。
  • 计算逻辑实现:在源文件中重写算子类中的Compute函数,实现算子计算逻辑。

下文将对这两部分的实现做具体说明。

如果您想获取AI CPU算子的实现样例,请参考表1

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词