更新时间:2021-03-18 GMT+08:00
分享

ImplyType函数

函数功能

设置算子执行方式。

函数原型

OpRegistrationData& ImplyType(domi::ImplyType imply_type);

参数说明

参数

输入/输出

说明

imply_type

输入

算子执行方式。

enum class ImplyType : unsigned int

{

BUILDIN = 0,// 内置算子,由OME正常执行

TVM, // 编译成tvm bin文件执行

CUSTOM, // 由用户自定义计算逻辑,通过CPU执行

AI_CPU, // AICPU 自定义算子类型

INVALID = 0xFFFFFFFF,

};

分享:

    相关文档

    相关产品