简介
本类是AI CPU算子的Kernel注册类，用于AI CPU框架侧获取算子Kernel的Compute函数相关信息，以及执行算子Kernel的Compute函数。
需要包含的头文件
1 | #include <cpu_kernel_register.h> |
Public成员函数
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)
父主题： CpuKernelRegister类