设置算子执行方式。
1
OpRegistrationData &ImplyType(const domi::ImplyType &imply_type)
参数
输入/输出
说明
imply_type
输入
算子执行方式。
1 2 3 4 5 6 7 8
enum class ImplyType : unsigned int { BUILDIN = 0,// 内置算子，由OME正常执行 TVM, // 编译成tvm bin文件执行 CUSTOM, // 由用户自定义计算逻辑，通过CPU执行 AI_CPU, // AI CPU 自定义算子类型 INVALID = 0xFFFFFFFF, };