Define REGISTER_KERNEL
Defined in File register_kernel.h
Define Documentation
-
REGISTER_KERNEL(arch, provider, data_type, op_type, creator)
Defined registering macro to register ordinary op kernel, which called by user directly.
- Parameters
arch – [in] Define deviceType, such as CPU.
provider – [in] Define the identification of user.
data_type – [in] Define kernel’s input data type.
op_type – [in] Define the ordinary op type.
creator – [in] Define a function pointer to create a kernel.