REGISTER_CPU_KERNEL
宏功能
注册kernel实现。
宏原型
REGISTER_CPU_KERNEL(type, clazz)
参数说明
- type:算子的OpType
- clazz:算子的kernel实现名称,即定义的自定义算子类的名称,自定义算子类为基类CpuKernel的派生类。
调用示例
namespace { const char *RESHAPE = "Reshape"; } namespace aicpu { uint32_t ReshapeCpuKernel::Compute(CpuKernelContext &ctx) { // 计算实现 } REGISTER_CPU_KERNEL(RESHAPE, ReshapeCpuKernel); } // namespace aicpu
父主题: 宏定义