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.

参数

• 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.