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.