设置算子执行方式。
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, };