文档
注册
评分
提单
论坛
小AI

简介

开发人员完成自定义算子的实现代码后,需要进行适配插件的开发将基于第三方框架的算子映射成适配昇腾AI处理器的算子,可调用REGISTER_CUSTOM_OP宏实现算子转换。在调用REGISTER_CUSTOM_OP宏时,以REGISTER_CUSTOM_OP开始,以“.”链接FrameworkType、OriginOpType、ParseParamsFn等接口。

例如:

REGISTER_CUSTOM_OP("OpType")    
   .FrameworkType(TENSORFLOW)     
   .OriginOpType("OriginOpType")    
   .ParseParamsByOperatorFn(ParseParamFunc)
   .ImplyType(ImplyType::TVM);

您可以在CANN软件安装后文件存储路径下的“include/register/register.h”文件中查看接口定义。

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

当前产品无相关内容

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