文档
注册

简介

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

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

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

如果您想获取AI CPU算子的实现样例,请参考表3。如果您想获取AI CPU算子的运行验证样例,请参考样例列表中的acl_execute_addblockcust、acl_execute_reshape样例。

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

当前产品无相关内容

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