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