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,
};
|
Parent topic: OpRegistrationData