Overview
The implementation of an AI CPU operator consists of:
- Operator class declaration: Declare the operator class in the header file. The custom operator class needs to inherit the CpuKernel base class.
- Compute logic implementation: Rewrite the Compute function in the operator class in the source file to implement the operator compute logic.
The following sections describe the implementation of the two parts in detail.
For details about how to obtain implementation samples of AI CPU operators, see Table 3. For details about how to obtain the running verification samples of AI CPU operators, see the acl_execute_addblockcust and acl_execute_reshape samples in Sample List.
Parent topic: Operator Code Implementation (AI CPU)