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类