开发人员完成自定义算子的实现代码后,需要进行适配插件的开发将基于第三方框架的算子映射成适配昇腾AI处理器的算子,可调用REGISTER_CUSTOM_OP宏实现算子转换。在调用REGISTER_CUSTOM_OP宏时,以REGISTER_CUSTOM_OP开始,以“.”链接FrameworkType、OriginOpType、ParseParamsFn等接口。
例如:
1 2 3 4 5 | REGISTER_CUSTOM_OP("OpType")
.FrameworkType(TENSORFLOW)
.OriginOpType("OriginOpType")
.ParseParamsByOperatorFn(ParseParamFunc)
.ImplyType(ImplyType::TVM);
|