Updated on 2022-03-13 GMT+08:00

ImplyType

Function

Sets the operator execution mode.

Syntax

OpRegistrationData& ImplyType(domi::ImplyType imply_type);

Parameter Description

Parameter

Input/Output

Description

imply_type

Input

Operator execution mode.

enum class ImplyType : unsigned int

{

BUILDIN = 0,// Built-in operator, which is normally executed by the OME

TVM, // Executed after being compiled into a TVM binary file

CUSTOM, // The calculation logic is customized by the user and executed by the CPU.

AI_CPU, // AI CPU custom operator type

INVALID = 0xFFFFFFFF,

};