Define REGISTER_KERNEL

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.