简介
自定义算子的构图编译接口,当算子进入GE构图编译流程后,若实现该接口,GE会回调Compile接口,完成算子编译相关处理。
需要包含的头文件
1 | #include <graph/custom_op.h> |
Public成员函数
virtual graphStatus Compile(gert::OpCompileContext *ctx) = 0
父主题: CompilableOp
自定义算子的构图编译接口,当算子进入GE构图编译流程后,若实现该接口,GE会回调Compile接口,完成算子编译相关处理。
1 | #include <graph/custom_op.h> |
virtual graphStatus Compile(gert::OpCompileContext *ctx) = 0