Overview
This class is the Kernel registration class of the AI CPU operator, which is used by the AI CPU framework to obtain the Compute function information of the operator kernel and execute the Compute function of the operator kernel.
Header Files to Be Included
1 | #include <cpu_kernel_register.h> |
Public Member Functions
static CpuKernelRegister &Instance() std::shared_ptr<CpuKernel> GetCpuKernel(const std::string &opType) std::vector<std::string>GetAllRegisteredOpTypes() const uint32_t RunCpuKernel(CpuKernelContext &ctx) uint32_t RunCpuKernelAsync(CpuKernelContext &ctx, const uint8_t wait_type, const uint32_t wait_id, std::function<uint32_t()> cb)
Parent topic: Class CpuKernelRegister