注册算子的原型定义,从而确保算子能够被框架正确识别、编译和执行。
算子原型主要描述了算子的输入输出、属性等信息以及算子在AI处理器上相关实现信息,并关联tiling实现等函数。算子原型通过自定义的算子类来承载,该算子类继承自OpDef类。完成算子的原型定义等操作后,需要调用OP_ADD接口,传入算子类型(自定义算子类的类名),进行算子原型注册。详细内容请参考算子原型定义。
1
|
OP_ADD(opType) |
|
参数 |
输入/输出 |
说明 |
|---|---|---|
|
opType |
输入 |
算子类型名称 |
无
无