ImplyType

函数功能

设置算子执行方式。

函数原型

OpRegistrationData &ImplyType(const domi::ImplyType &imply_type)

参数说明

参数

输入/输出

说明

imply_type

输入

算子执行方式。

enum class ImplyType : unsigned int
{
BUILTIN = 0, // 内置算子,由OME正常执行
BUILDIN = BUILTIN, // 兼容历史拼写错误版本,请使用BUILTIN
TVM,        // 编译成tvm bin文件执行
CUSTOM,     // 由用户自定义计算逻辑,通过CPU执行
AI_CPU,      // AI CPU 自定义算子类型
CCE,        // Cce
GELOCAL,    // GE local, do not need execute by device
HCCL,       // Hccl
INVALID = 0xFFFFFFFF,
};