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.