Define REGISTER_CUSTOM_KERNEL
Defined in File register_kernel.h
Define Documentation
-
REGISTER_CUSTOM_KERNEL(arch, provider, data_type, op_type, creator)
Defined registering macro to register custom 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 concrete type of a custom op.
creator – [in] Define a function pointer to create a kernel.