ImplyType

Description

Sets the operator execution mode.

Prototype

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

Parameters

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 built into a TVM binary file.
CUSTOM,     // The compute logic is user-defined and executed by CPU.
AI_CPU,      // Custom AI CPU operator.
INVALID = 0xFFFFFFFF,
};