ImplyType
函数功能
设置算子执行方式。
函数原型
OpRegistrationData& ImplyType(const domi::ImplyType &imply_type)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
imply_type |
输入 |
算子执行方式。 enum class ImplyType : unsigned int { BUILDIN = 0,// 内置算子,由OME正常执行 TVM, // 编译成tvm bin文件执行 CUSTOM, // 由用户自定义计算逻辑,通过CPU执行 AI_CPU, // AI CPU 自定义算子类型 INVALID = 0xFFFFFFFF, }; |
父主题: OpRegistrationData类